三、Pytorch环境安装
基本按照教程来,就没什么问题。
Pytorch安装
wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.8.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
pip3 install Cython
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl
在安装时,长期得不到响应,估计是因为外网的原因,得挂梯子才行。可以先将该文件下载,进行本地安装。直接运行命令
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl
等待一会儿,安装完成:
简单的检验下安装是否成功:
Torchvision安装
官方的教程:
$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch <version> https://github.com/pytorch/vision torchvision # see below for version of torchvision to download
$ cd torchvision
$ export BUILD_VERSION=0.x.0 # where 0.x.0 is the torchvision version
$ python3 setup.py install --user
$ cd ../ # attempting to load torchvision from build dir will result in import error
$ pip install 'pillow<7' # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6
我呢,对照官方的教程,对应的改下版本号就可以了,官方给出了不同版本Pytorch与对应的Torchvision版本号。
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision # 我的pytorch对应的是0.9.0的torchvision
cd torchvision
export BUILD_VERSION=0.9.0
python3 setup.py install --user
耗时挺久的,安装完成:
简单的检验下安装是否成功:
在import torchvision
后,出现过AttributeError: module 'PIL.Image' has no attribute 'BILINEAR’
的问题,是由于pillow版本的问题,先卸载当前的版本,安装一个较低版本即可。