安装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

下载之后进行安装：