在jetson nano中安装Anaconda、pytorch 以及 torchvision
1.Pytorch
下载安装
- 首先登录英伟达官网下载Pytorch安装包,这里以PyTorch v1.10.0为例
- 安装依赖库
c
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev libopenblas-base libopenmpi-dev
- 安装Pytorch
c
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是否安装成功
c
python3
import torch
x = torch.rand(5, 3)
print(x)
- 查看版本信息
c
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
c
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版本太高,卸载重装
cpp
sudo pip3 uninstall pillow
sudo pip3 install pillow
- 查看版本信息
cpp
import torchvision
print(torchvision.__version__)