RuntimeError: Detected that PyTorch and torchvision were compiled with different CUDA versions. PyTo

RuntimeError: Detected that PyTorch and torchvision were compiled with different CUDA versions. PyTorch has CUDA Version=11.8 and torchvision has CUDA Version=11.7. Please reinstall the torchvision that matches your PyTorch install.

根据错误信息,您安装的 PyTorch 和 torchvision 版本不兼容,因为它们编译时使用了不同的 CUDA 版本。PyTorch 使用的 CUDA 版本是 11.8,而 torchvision 使用的 CUDA 版本是 11.7。

要解决这个问题,有几个可能的解决方法:

  1. 升级或降级 PyTorch:您可以尝试升级或降级 PyTorch,使其与已安装的 torchvision 版本兼容。您可以在 PyTorch 官方文档中找到与 torchvision 版本兼容的 PyTorch 版本,并重新安装 PyTorch。

  2. 重新安装 torchvision:您也可以尝试卸载当前的 torchvision,并重新安装与您当前的 PyTorch 版本兼容的新版本。您可以使用以下命令卸载 torchvision:

    pip uninstall torchvision
    

    然后,根据您的 PyTorch 版本重新安装 torchvision。例如,如果您的 PyTorch 版本是 1.9.0,则可以使用以下命令安装相应版本的 torchvision:

    pip install torchvision==0.10.0
    

请注意,具体的版本号可能会因您当前环境和需要而不同。请根据您的环境和需求选择正确的版本进行安装。

总之,确保您所使用的 PyTorch 和 torchvision 版本兼容,并按照上述方法之一进行处理,应该能够解决 CUDA 版本不兼容的问题。

要通过 pip 安装与 CUDA 11.8 兼容的 torchvision,您可以使用以下命令:

pip install torch==1.10.0+cu118 torchvision==0.11.1+cu118 -f https://download.pytorch.org/whl/cu118/torch_stable.html

这将安装 CUDA 11.8 版本的 PyTorch(1.10.0)和对应版本的 torchvision(0.11.1),并且从 PyTorch 官方源中下载安装文件。

请注意,命令中的 cu118 表示 CUDA 11.8 版本,如果您使用的是其他 CUDA 版本,请将其替换为对应的 CUDA 版本号。

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

irrationality

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值