安装pytorch---- 基于Jetson Xavier nx

1.前言

Jetson Xavier nx 安装pytorch 需要到Nvidia 官网下载，而不是去pytorch下载。

pytorch官网：https://pytorch.org/get-started/previous-versions/#osx-14

nvidia-pytorch官网 ：https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048

2.环境信息

Version：3.1.4
Jetpack: 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-dev
sudo apt-get install liblapack-dev libblas-dev gfortran

sudo apt-get install libopenblas-base libopenmpi-dev
pip3 install mpi4py
pip3 install Cython

4.安装pytorch

下载编译好的.whl文件, 官网下载地址：
https://forums.developer.nvidia.com/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-dev
git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision   #更改对应版本，这里对应的是0.9.0
cd torchvision
export BUILD_VERSION=0.9.0  
python3 setup.py install --user

官网可以查看对应的Torchvision版本

验证：

>>> import torchvision
>>> print(torchvision.__version__)

查看cuda是否可用：

import torch
print(torch.cuda.is_available())  >>>返回 True

参考链接：Jetson Xavier nx安装pytorch
PyTorch for Jetson