树莓派安装torch与torchvision测试
如使用以下命令无法安装,则考虑后面的安装方式
点击查看代码
pip install torch==1.8.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install torchvision==0.9.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
在官方下载页搜索下载
cpu/torch-1.8.1-cp39-cp39-manylinux2014_aarch64.whl
cpu/torchvision-0.9.1-cp39-cp39-manylinux2014_aarch64.whl
注意版本对应,且PyTorch 仅能在 64 位Raspberry Pi (树莓派)上安装,32位的不支持
cd进入文件目录,安装
pip3 install torch-1.8.1-cp39-cp39-manylinux2014_aarch64.whl
pip3 install torchvision-0.9.1-cp39-cp39-manylinux2014_aarch64.whl
安装完后import测试