Torchvision环境搭建
Torchvision环境搭建1、系统信息2、安装依赖3、安装Torchvision3.1、离线安装3.2、在线安装3.3、源码编译4、验证安装参考资料
1、系统信息
2、安装依赖
sudo apt update xxxxxxxxxx sudo apt install ninja-build libwebp-dev libjpeg-dev -y
3、安装Torchvision
根据PyTorch与Torchvision版本对应关系，我们需要安装v0.20版本Torchvision
| torch | torchvision | Python |
|---|---|---|
main / nightly | main / nightly | >=3.9, <=3.12 |
2.5 | 0.20 | >=3.9, <=3.12 |
3.1、离线安装
手动进入下载网址下载最新的Torchvision版本。
xxxxxxxxxx cd Downloads/ xxxxxxxxxx sudo pip3 install torchvision-0.20.0a0+afc54f7-cp310-cp310-linux_aarch64.whl
3.2、在线安装
xxxxxxxxxx sudo pip3 install https://github.com/ultralytics/assets/releases/download/v0.0.0/torchvision-0.20.0a0+afc54f7-cp310-cp310-linux_aarch64.whl
3.3、源码编译
手动下载v0.20版本：
xxxxxxxxxx https://github.com/pytorch/vision/tree/v0.20.0
进入github下载的解压文件夹内编译源码：
xxxxxxxxxx cd vision-0.20.0 xxxxxxxxxx sudo python3 setup.py install
4、验证安装
xxxxxxxxxx python3 -c "import torchvision; print(f'Torchvision: {torchvision.__version__}')"
