PyTorch与Torchvision CUDA编译版本不匹配,多次重装仍无解求助
解决PyTorch与Torchvision CUDA版本不匹配问题
1. 彻底清理现有安装残留
- 先卸载所有相关包:
pip uninstall -y torch torchvision torchaudio - 找到Python的
site-packages目录,手动删除torch/、torchvision-*.dist-info/等相关残留文件夹 - 清理pip缓存避免旧安装包干扰:
pip cache purge
2. 确认本地CUDA驱动兼容性
- 先检查系统CUDA驱动支持的最高版本:
驱动支持的CUDA版本需≥PyTorch使用的11.6,若版本不足需先更新显卡驱动nvidia-smi
3. 精准匹配版本安装
- 直接指定对应CUDA版本的安装包,避免默认安装的版本偏差。以PyTorch 11.6为例,执行对应命令(根据自身Python版本调整,如
cp38对应Python3.8):
版本对应规律:torchvision版本号通常比PyTorch小0.1(如torch 2.0+cu118对应torchvision 0.15+cu118)pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
4. 验证安装结果
- 运行以下代码确认版本一致性:
输出的两个版本需完全一致import torch import torchvision print(f"PyTorch CUDA版本: {torch.version.cuda}") print(f"Torchvision编译CUDA版本: {torchvision._C._CUDA_VERSION}")
5. 排查虚拟环境问题
- 确认当前激活的是目标虚拟环境,避免安装到全局Python环境中
- 若旧虚拟环境存在异常,直接删除后重新创建干净的虚拟环境再执行安装操作
内容的提问来源于stack exchange,提问作者Alan Carbajo
相关产品推荐
相关产品推荐

