首先切换为国内软件源,然后执行更新操作
sudo apt-get update
sudo apt-get upgrade
安装torch的依赖
sudo apt-get install libopenblas-base libopenmpi-dev
安装 sudo apt-get install python3-pip
下载官网torch
PyTorch for Jetson - version 1.10 now available - Jetson Nano - NVIDIA Developer Forums
Jetson nano自带2.7和3.6版本的python 自带 cuda10.2
我这下载的是 python3支持 的 1.8.0版本
进入下载.whl的目录,由于pip3 执行下面的命令安装Cpython和使用本地whl包安装torch,注意文件名中的版本要对应。
pip3 install Cython
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl
打印 Successfully installed torch-1.8.0
进入python3, 执行代码import torch,如果没有报错则说明torch 即安装成功
安装torchvision
使用git获取torchvision的源码,注意pytorch和torchvision的版本一定要对应
git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision
编译torchvision
接下来就是编译阶段了,进入torchvision目录,输入命令编译,大致需要花费十分钟。
在编译之前 使用pip/pip3安装 pillow。
pip3 install 'pillow<7'
下载速度比较慢时,可以使用国内的豆瓣源下载安装,命令如下所示。
pip3 install -i http://pypi.douban.com/simple --trusted-host pypi.douban.com 'pillow<7'
cd torchvision
export BUILD_VERSION=0.x.0 #将x改成自己的版本
sudo python3 setup.py install --user