1、首先查看设备的cuda版本
nvcc -V
cat /usr/local/cuda/version.txt
2、根据cuda版本选择合适的pytorch,下载pytorch。
Cuda9.1的pytorch下载地址如下
https://download.pytorch.org/whl/cu91/torch/
pip install https://download.pytorch.org/whl/torch-1.12.0-cp37-cp37m-manylinux2014_aarch64.whl
3、出现报错,设备不支持版本。
ERROR: torch-1.12.0-cp37-cp37m-manylinux2014_aarch64.whl is not a supported wheel on this platform.
使用命令行查看设备支持的版本。
pip debug --verbose
设备支持的版本的如下。