树莓派64位系统安装pytorch & torchvision

Miniconda安装

Miniconda源链接

测试发现很多版本安装会出现Illegal instruction错误，不建议安装大于4.9的版本。
顺利安装 Miniconda3-py37_4.9.2-Linux-aarch64.sh

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-py37_4.9.2-Linux-aarch64.sh

关闭终端，重开终端。默认base环境自动启动，关闭自启动

conda config --set auto_activate_base false 

新建环境为torch安装做准备

conda create -n torch python=3.7

新环境默认安装pip=21.1.2，当使用pip命令时，会出现illegal instruction，将pip版本退回到21.0.1解决。

经测试使用以下方式安装的torch无法正常运行，原因未知。

conda install pytorch

删除环境更换安装方式

conda uninstall -n 环境名 --all

pytorch安装

torch/torchvison源链接

wget https://github.com/KumaTea/pytorch-aarch64/releases/download/v1.10.0/torch-1.10.0-cp37-cp37m-linux_aarch64.whl

然后

pip install torch-1.10.0-cp37-cp37m-linux_aarch64.whl

安装numpy

pip install numpy

检查确认安装情况

(torch) pi@robupi:~ $ python
Python 3.7.10 (default, Mar 15 2021, 20:52:10)
[GCC 10.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> exit()

torchvison安装

torchvison安装受pytorch版本限制，版本对应关系

wget https://github.com/KumaTea/pytorch-aarch64/releases/download/v1.10.0/torchvision-0.11.0-cp37-cp37m-linux_aarch64.whl

安装torchvison

pip install torchvision-0.11.0-cp37-cp37m-linux_aarch64.whl

检查安装情况

(torch) pi@robupi:~ $ python
Python 3.7.10 (default, Mar 15 2021, 20:52:10)
[GCC 10.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torchvision
>>> torchvision.__version__
'0.11.0'