树莓派安装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测试

posted @ 2022-11-29 22:40  Dapenson  阅读(602)  评论(0编辑  收藏  举报