Jetson nano安装torch和torchvision

前言：首先下载torch和torchvision

百度网盘（torch1.7.0 torchvision 0.8.0）

链接:https://pan.baidu.com/s/15d0SGz1AX1dLAfDmCpw9OA?pwd=nvvn 
提取码:nvvn

1、首先我是在archiconda环境中安装的torch和torchvision，所以首先创建一个虚拟环境

conda create -n torch-1.7.0 python=3.6

conda activate torch-1.7.0

2、安装torch依赖

sudo apt-get install libopenblas-base libopenmpi-dev 

3、安装pip3

sudo apt-get install python3-pip 

4、下载官网torch，前面已下载

5、进入前面下载.whl 的目录，由于pip3 执行下面的命令安装Cpython和使用本地whl包安装torch，注意文件名中的版本要对应。

pip3 install Cython
pip3 install numpy torch-1.7.0-cp36-cp36m-linux_aarch64.whl

 安装完成后会打印输出Successfully installed dataclasses-0.8 future-0.18.2 numpy-1.19.5 torch-1.7.0 typing-extensions-4.1.1

6、验证torch是否安装成功

python3

import torch

torch.cuda.is_

如果没有报错或输出true，说明torch安装成功

7、torchvision安装

pip3 install torchvision-0.8.0a0+291f7e2-cp36-cp36m-linux_aarch64.whl

安装成功会提示Successfully installed pillow-8.4.0 torchvision-0.8.0a0+291f7e2

8、验证torchvision是否安装成功

pip3 list

python

import torchvision

torchvision.__version__ 

 此时会输出torchvision的版本号

9、运行代码

cd 工程文件

python app.py或者train.py

接下来会提示好多模块没有安装，一个一个安装就行，可通过

pip install 模块名 -i https://pypi.tuna.tsinghua.edu.cn/simple 

注意：比较难以安装的是matplotlib，大概需要花费20分钟，所以要有耐心，尤其是会在

Building wheels for matplotlib(set.py) . . .处停顿好长时间，一定要有耐心。安装成功会显示

Successfully installed cycler-0.11.0 kiwisolver-1.3.1 matplotlib-3.3.4 pyparsing-3.0.9