我和jetson-Nano的故事（12）——安装pytorch 以及 torchvision

本文主要是介绍我和jetson-Nano的故事（12）——安装pytorch 以及 torchvision，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

1.Pytorch下载安装

1. 首先登录英伟达官网下载Pytorch安装包，这里以PyTorch v1.10.0为例

2. 安装依赖库

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev libopenblas-base libopenmpi-dev

3. 安装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的）

4. 验证Pytorch是否安装成功

python3
import torch
x = torch.rand(5, 3)
print(x)

5. 查看版本信息

import torch
print(torch.__version__)

2.Torchvision安装

Torchvision版本要与Pytorch版本相匹配,我们前面安装的Pytorch版本为1.10.0，Torchvision安装 v0.10.0版本

1. Torchvision版本要与Pytorch版本相匹配,我们前面安装的Pytorch版本为1.10.0，Torchvision安装 v0.11.0版本
   
2. 下载、安装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

3. 查看版本信息

import torchvision
print(torchvision.__version__)

这篇关于我和jetson-Nano的故事（12）——安装pytorch 以及 torchvision的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---