You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

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驱动支持的最高版本:
    nvidia-smi
    
    驱动支持的CUDA版本需≥PyTorch使用的11.6,若版本不足需先更新显卡驱动

3. 精准匹配版本安装

  • 直接指定对应CUDA版本的安装包,避免默认安装的版本偏差。以PyTorch 11.6为例,执行对应命令(根据自身Python版本调整,如cp38对应Python3.8):
    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
    
    版本对应规律:torchvision版本号通常比PyTorch小0.1(如torch 2.0+cu118对应torchvision 0.15+cu118)

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

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.08.12 19:31:03