Jetson AGX Xavier配置PyTorch与TorchVision流程

Jetson AGX Xavier配置PyTorch与TorchVision流程

1 环境

Jetson AGX Xavier，JetPack 4.4 production release (L4T R32.4.3)，Python 3.6.9。

 

需要安装的PyTorch与TorchVision对应的版本号为：

PyTorch v1.6 + torchvision v0.7.0-rc2

 

PyTorch与JetPack匹配的版本号：

 

PyTorch与TorchVision匹配的版本号：

2 安装PyTorch

PyTorch v1.6的wheels下载链接为：

https://nvidia.box.com/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl

 

安装命令为：

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev

pip3 install Cython

pip3 install numpy ~/下载/torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl

3 安装TorchVision

安装命令为：

sudo apt-get install libjpeg-dev zlib1g-dev

git clone --branch v0.7.0-rc2 https://github.com/pytorch/vision torchvision

cd torchvision

sudo python3 setup.py install

cd ../

4 测试PyTorch与TorchVision

输入下列命令：

python3

import torch

print(torch.__version__)

print('CUDA available: ' + str(torch.cuda.is_available()))

print('cuDNN version: ' + str(torch.backends.cudnn.version()))

a = torch.cuda.FloatTensor(2).zero_()

print('Tensor a = ' + str(a))

b = torch.randn(2).cuda()

print('Tensor b = ' + str(b))

c = a + b

print('Tensor c = ' + str(c))

import torchvision

print(torchvision.__version__)

 

安装成功后的测试结果如下图所示：

5 参考文献

1. https://forums.developer.nvidia.com/t/pytorch-for-jetson-nano-version-1-5-0-now-available/72048
2. 查看jetpack的版本：https://www.jianshu.com/p/7f2d8a563ded
3. Linux下查看python版本号：https://blog.csdn.net/huangzx3/article/details/80067417