安装 torchvision 的方法

torch与torchvision版本对应关系 

torch    torchvision    Python
main / nightly    main / nightly    >=3.9, <=3.12
2.5    0.20    >=3.9, <=3.12
2.4    0.19    >=3.8, <=3.12
2.3    0.18    >=3.8, <=3.12
2.2    0.17    >=3.8, <=3.11
2.1    0.16    >=3.8, <=3.11
2.0    0.15    >=3.8, <=3.11

安装指令：

cd ./Downloads
git clone --branch v0.16.2 https://github.com/pytorch/vision

2.2.3 编译安装torchvision

cd vision-0.16.2                    # 进入torchvision目录
export BUILD_VERSION=0.16.2          # 将BUILD_VERSION环境变量设置为值 0.16.2  
python3 setup.py install --user        # 使用 Python 的 setuptools 工具将vision包安装到当前用户的本地目录中

3. 验证

查看pytorch运行时真正调用的cuda、cudnn版本：

python -c "import torch; import torchvision; print('PyTorch version:', torch.__version__); print('CUDA available:', torch.cuda.is_available()); print('CUDA version:', torch.version.cuda); print('cuDNN enabled:', torch.backends.cudnn.enabled); print('cuDNN version:', torch.backends.cudnn.version()); print('Torchvision version:', torchvision.__version__)"