8、使用PyTorch

一、安装PyTorch

1、下载与安装

安装教程可以参考这个网址

wget https://nvidia.box.com/shared/static/wa34qwrwtk9njtyarwt5nvo6imenfy26.whl -O torch-1.7.0-cp36-cp36m-linux_aarch64.whl

因为国内无法访问到box.com,所以我们需要从其他的地方下载。网盘地址
密码：u509
1、针对python3.6版本

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

pip3 install Cython

pip3 install numpy torch-1.7.0-cp36-cp36m-linux_aarch64.whl

2、torchvision

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

cd torchvision
export BUILD_VERSION=0.8.1
sudo python setup.py install

2、验证是否安装成功

>>> 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_