超简单:解决yolo v5测试detect时torch与torchvision版本不匹配的问题

解决YOLO V5报错: Couldn’t load custom C++ ops. This can happen if your PyTorch and torchvision…

简单到你无法想象

在CSDN上面搜了很多解决方法,但是还是解决不了问题,他一直报错。后期我自己解决问题之后,发现有些博客其实已经给出了答案,但是非常模棱两可,不知道为什么程序调试的问题老是喜欢玩文字游戏,在哪里巴拉巴拉讲一堆原因就是没给出非常详细的解决方案。

前提:

  • 必须要确保你的GPU环境配置已经没有问题并且相关准备工作已经完成。
  • 只是因为你的torch和torchvision的版本无法匹配。
  • 关于torch和torchvision匹配的关系按照我的方法好像不用刻意去关注, 只需要确定好你的cuda对应的torch是哪个版本即可

解决步骤:

  • 首先进入pytorch官网,地址:link

  • 点击上方菜单栏"Get-started"
    在这里插入图片描述

  • 点击"Previous PyTorch Versions"
    在这里插入图片描述

  • 注意:千万不要选择使用conda安装的方式安装torch和torchvision

  • 根据你的cuda版本去选择pip版本的torch和torchvision(以我的V11.3为例)

  • 在这里插入图片描述

  • 找到适合你的pip安装命令之后,打开Anaconda自带的终端,打开你创建的虚拟环境,然后把符合你版本安装的pip命令复制回车即可。

    • 这一步需要注意:一定是要在打开虚拟环境下进行(因为我是在虚拟环境下安装的,也许有其他方法,但我不确定,所以如果和我一样使用虚拟环境安装的uu可以注意一下这点)
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值