在树莓派 4 上安装 torchvision

Torchvision 是常用数据集、架构和图像算法的集合。当您使用我们在 GitHub 上找到的一个wheel时，安装很简单。您还可以从头开始构建 torchvision。在这种情况下，您只需从官方 GitHub 页面下载您选择的版本，在setup.py的第 32 行修改版本号并发出命令$ python3 setup.py bdist_wheel。

Torchvision 假设 PyTorch 安装在你的机器上正手。

V0.12.0

Used with PyTorch 1.10.0
# the dependencies
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev
sudo apt-get install libavcodec-dev libavformat-dev libswscale-dev
# install gdown to download from Google drive, if not done yet
sudo -H pip3 install gdown

Buster OS
# download TorchVision 0.12.0
gdown https://drive.google.com/uc?id=1uQgjrucNSfpliGBWeKUOuo4XGWiBaxQx
# install TorchVision 0.12.0
sudo -H pip3 install torchvision-0.12.0a0+09c5ddd-cp37-cp37m-linux_aarch64.whl
# clean up
rm torchvision-0.12.0a0+09c5ddd-cp37-cp37m-linux_aarch64.whl

or Bullseye OS
# download TorchVision 0.12.0
gdown https://drive.google.com/uc?id=1nX4uSoubwq8NB9k-2DF6LbPyiqwocgEW
# install TorchVision 0.12.0
sudo -H pip3 install torchvision-0.12.0a0+57a77c4-cp39-cp39-linux_aarch64.whl
# clean up
rm torchvision-0.12.0a0+57a77c4-cp39-cp39-linux_aarch64.whl

v0.10.0

Used with PyTorch 1.9.0
# the dependencies
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev
sudo apt-get install libavcodec-dev libavformat-dev libswscale-dev
# install gdown to download from Google drive, if not done yet
sudo -H pip3 install gdown

Buster OS
# download TorchVision 0.10.0
gdown https://drive.google.com/uc?id=1XyYG6hHh7QYaLrvE-mByvhNK_0rLOd0G
# install TorchVision 0.10.0
sudo -H pip3 install torchvision-0.10.0a0+300a8a4-cp37-cp37m-linux_aarch64.whl
# clean up
rm torchvision-0.10.0a0+300a8a4-cp37-cp37m-linux_aarch64.whl

or Bullseye OS
# download TorchVision 0.10.0
gdown https://drive.google.com/uc?id=13fKq1V-zLbLBWPw3zP4EZxyMXyOiJWPi
# install TorchVision 0.10.0
sudo -H pip3 install torchvision-0.10.0a0+300a8a4-cp39-cp39-linux_aarch64.whl
# clean up
rm torchvision-0.10.0a0+300a8a4-cp39-cp39-linux_aarch64.whl

安装后，您可能希望通过验证发布版本来检查 torchvision。