在jetson nano中安装Anaconda、pytorch 以及 torchvision
- 1.`Pytorch`下载安装
- 2.`Torchvision`安装
1.Pytorch下载安装
- 首先登录英伟达官网下载Pytorch安装包，这里以PyTorch v1.10.0为例
- 安装依赖库
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev libopenblas-base libopenmpi-dev
- 安装Pytorch
sudo chmod a+x torch-1.10.0-cp36-cp36m-linux_aarch64.whl
sudo pip3 install torch-1.10.0-cp36-cp36m-linux_aarch64.whl
安装成功（注意Python版本得是3.6的）
- 验证Pytorch是否安装成功
python3
import torch
x = torch.rand(5, 3)
print(x)
- 查看版本信息
import torch
print(torch.__version__)
2.Torchvision安装
Torchvision版本要与Pytorch版本相匹配,我们前面安装的Pytorch版本为1.10.0，Torchvision安装 v0.10.0版本
- Torchvision版本要与Pytorch版本相匹配,我们前面安装的Pytorch版本为1.10.0，Torchvision安装 v0.11.0版本
- 下载、安装torchvision
git clone --branch v0.11.0 https://github.com/pytorch/vision torchvision
cd torchvision
export BUILD_VERSION=0.11.0
sudo python3 setup.py install
出现报错可能是Pillow版本太高,卸载重装
sudo pip3 uninstall pillow
sudo pip3 install pillow
- 查看版本信息
import torchvision
print(torchvision.__version__)
