1.前言
Jetson Xavier nx 安装pytorch 需要到Nvidia 官网下载，而不是去pytorch下载。
pytorch官网：/get-started/previous-versions/#osx-14
nvidia-pytorch官网 ：https://forums./t/pytorch-for-jetson/72048
2.环境信息
Version：3.1.4Jetpack: 4.4.1 [L4T 32.4.4]CUDA: 10.2.89
3.安装依赖
sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-devsudo apt-get install liblapack-dev libblas-dev gfortransudo apt-get install libopenblas-base libopenmpi-devpip3 install mpi4pypip3 install Cython
4.安装pytorch
下载编译好的.whl文件, 官网下载地址：
https://forums./t/pytorch-for-jetson/72048
下载之后进行安装：
pip3 install torch-1.8.0-cp36-cp36m-linux_aarch64.whl
其中.whl文件下载特别慢，最好爬墙！！
验证：
>>> import torch>>> print(torch.__version__)
5. 安装Torchvision
sudo apt-get install libjpeg-dev zlib1g-devgit clone --branch v0.9.0 /pytorch/vision torchvision #更改对应版本，这里对应的是0.9.0cd torchvisionexport BUILD_VERSION=0.9.0 python3 setup.py install --user
官网可以查看对应的Torchvision版本
验证：
>>> import torchvision>>> print(torchvision.__version__)
查看cuda是否可用：
import torchprint(torch.cuda.is_available()) >>>返回 True
参考链接：Jetson Xavier nx安装pytorch
PyTorch for Jetson
如果觉得《安装pytorch---- 基于Jetson Xavier nx》对你有帮助，请点赞、收藏，并留下你的观点哦！
