本人是第一次使用jeston nano，下载torch和torchvision碰了很多壁，因此这里把经验教训写下来，希望能帮助到像我一样的小白，让大家少走点弯路，由于第一次在csdn编辑，格式不美观，有错误见谅~
方式一：通过离线包安装
1.安装jtop，查看jetpack版本
pip3 install jetson-stats
安装好jtop后，输入：
jtop
在图中可以看到我的jetpack版本是 4.2，cuda版本是10.0（cuda的安装在网上有许多教程），后续安装torch和torchvision需要根据这个jetpack版本和python版本来，否则会报错
OSError: libcudart.so.10.0: cannot open shared object file: No such file
在英伟达官网查找jetpack对应torch版本，以下贴出我在官网查询的结果
PyTorch for Jetson - Jetson & Embedded Systems / Jetson Nano - NVIDIA Developer Forums
可以查到jetpack4.2和python3.6支持torch从1.0.0到1.4.0的版本，torch可以直接从英伟达官网下载，但是torchvision要使用pip或者conda命令，pip命令要涉及到使用清华镜像源，我这里一开始选择的是1.4.0的版本，但是涉及到后续安装torchvision从清华镜像源下载没有对应的torchvision版本，清华镜像源对应的版本有如下 ：
所以为了能下载到对应版本的torchvision，我这里选择下载torchv1.0.0，对应的torchvision版本是0.2.2
直接点击下载torch，下载好后，通过ssh传输到nano
cd （torch所在目录）
pip3 install "torch-1.0.0a0+bb15580-cp36-cp36m-linux_aarch64.whl" #注意区别pip3和pip
python3.6
import torch
print(torch.cuda.is_available())
可以看到成功的调用了GPU，说明torch安装成功
然后使用命令安装torchvision0.2.2
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torchvision==0.2.0
等待安装完毕，同样在python3.6验证torchvision是否安装成功
python3.6
import torchvision
没有报错，说明安装成功
方式二：
如果可以ping通github的网站或者可以科学上网，那么可以直接用英伟达官网的教程，直接在命令行窗口使用命令安装，具体教程如下：
（->python3.6，torch安装）
wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.8.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev pip3 install Cython
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl
->torchvision
$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch <version> https://github.com/pytorch/vision torchvision # see below for version of torchvision to download
$ cd torchvision
$ export BUILD_VERSION=0.x.0 # where 0.x.0 is the torchvision version
$ python3 setup.py install --user
$ cd ../ # attempting to load torchvision from build dir will result in import error
$ pip install 'pillow<7' # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6
到此torch和torchvision就全部安装完毕了。
可以跑一个简单的深度神经网络的模型试一下
在经过十七次迭代后，准确率已经达到了0.93以上。
评论(0)