Ubuntu:JETSON NX配置环境,安装pytorch,torchvision避坑总结

1.安装python以及依赖包

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install git cmake python3-dev
sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev

备注:上面的步骤一步一步执行

2.配置python环境

sudo apt-get install python3-pip

sudo pip3 install -U pip testresources setuptools

3.安装Pytorch:

sudo apt-get install libopenblas-base libopenmpi-dev

sudo pip3 install mpi4py

sudo pip3 install Cython

备注:先下载好pytorch编译好的.whl文件,在Jetson Nano部署主要难点是pytorch安装,使用Nvidia官方预编译whl文件.选择自己需要的版本下载.

下载链接:PyTorch for Jetson - version 1.10 now available - Jetson Nano - NVIDIA Developer Forums

不需要解压,下载完成拷贝到NX后,进行安装,注意文件存放位置和执行命令的位置相一致,输入命令:

sudo pip3 install torch-1.7.0-cp36-cp36m-linux_aarch64.whl

4.安装torchvision

进入其官方GitHub:GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision
找到自己需要的版本下载下来,torch和torchvision的版本需要相互对应,比如博主最终安装的是Pytorch1.7.0,所以torchvision版本选择0.8.

 下载完成后,解压,进入解压后的目录:

 在此处打开终端,输入:

peter@peter-desktop:~/Desktop/jetson-nano-fire-detection-master/vision-0.8.0$  sudo python3 setup.py install

如果没有报错,那么恭喜你,安装成功啦!

5.[避坑总结]torchvision安装报错

博主查阅诸多资料,看了很多的帖子都没有找到合适的解决方法 ,最后终于找到了一篇帖子完美的解决了博主的问题,在此非常感谢!

通过阅读报错的地方发现了问题所在,这一条报错信息:

/home/peter/Desktop/jetson-nano-fire-detection-master/vision-0.8.0/torchvision/csrc/cpu/decoder/defs.h:12:10: 

指向/home/peter/Desktop/jetson-nano-fire-detection-master/vision-0.8.0/torchvision/csrc/cpu/decoder/defs.h文件 

 根据这个地址找到错误出现的地方:

          

出现这个错误其实就是缺少这些包嘛,只需要去下载这些包就可以完美的解决问题啦!

sudo apt install libavcodec-dev
sudo apt install libavformat-dev
sudo apt install libswscale-dev

再到vision文件夹下打开终端,然后在终端中输入

sudo python3 setup.py install

 在执行这一行代码时发现pillow包一直没有安装成功,其实到这里torchvision已经安装成功了,输入命令pip3 list可以看到。只需要pip3 install pillow安装pillow包即可,如果下载速度慢,可以加入-i加清华源下载。

peter@peter-desktop:~/Desktop/jetson-nano-fire-detection-master/vision-0.8.0$ pip3 install pillow  -i https://pypi.tuna.tsinghua.edu.cn/simple

 

安装成功啦! 


 

猜你喜欢

转载自blog.csdn.net/JulyLH/article/details/123140461