Linux Ubuntu下Anaconda环境无法安装GPU版torchvision 0.12.0求方案

1. 确认版本兼容性 #

torchvision与PyTorch版本严格绑定，torchvision 0.12.0必须搭配PyTorch 1.11.0使用。先检查当前Anaconda环境中的PyTorch版本：

python -c "import torch; print(torch.__version__)"

若版本不符，需先同步PyTorch版本，否则会出现兼容性问题。

2. 激活目标Anaconda环境 #

切换到你的目标环境（替换your_env_name为实际环境名）：

conda activate your_env_name

3. 使用conda安装GPU版torchvision 0.12.0 #

根据你的CUDA版本选择对应指令：

• 若CUDA版本为11.3：

conda install -c pytorch torchvision=0.12.0 torch=1.11.0 cudatoolkit=11.3

• 若CUDA版本为11.1：

conda install -c pytorch torchvision=0.12.0 torch=1.11.0 cudatoolkit=11.1

• 若pytorch频道无对应包，可添加conda-forge频道：

conda install -c pytorch -c conda-forge torchvision=0.12.0 torch=1.11.0 cudatoolkit=你的CUDA版本号

4. 验证安装结果 #

执行以下命令确认torchvision版本及GPU可用性：

python -c "import torchvision; print(torchvision.__version__); print(torch.cuda.is_available())"

输出显示0.12.0和True即为安装成功。

注意事项 #

由于环境中包含NVIDIA Rapids，需确保安装的cudatoolkit版本与Rapids依赖的CUDA版本一致，避免库冲突。

内容的提问来源于stack exchange，提问作者gkasap