树莓派Raspberry Pi 4B 安装 PyTorch1.7.1

安装torch

pip3 install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 -f https://torch.maku.ml/whl/stable.html

检查是否安装成功

import torch
torch.version

若出现错误

RuntimeError: module compiled against API version 0xc but this version of numpy is 0xb

更新numpy

pip3 install --upgrade numpy

参考 树莓派Raspberry Pi 4B 安装 PyTorch