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》对你有帮助,请点赞、收藏,并留下你的观点哦!