ubuntu中 使用GPU运行yolov5 python detect.py时报错:非法指令(核心已转储)解决方法

ubuntu18.04

RTX 3060 

CUDA11.3

NVIDIA DRIVER 515

通过pytorch官方网站下载的CUDA11.3版本的pytorch和torchvision

 在yolov5中运行 python detect.py时报错

非法指令(核心已转储)

        

 

可能是因为torch版本太低造成

删除原来的torch

conda uninstall pytorch
conda uninstall libtorch

在官网中找到如下命令

选择自己的CUDA版本

        

执行后再次运行python detect.py

即可运行成功

 

        小白一枚 就这个问题搞了快一周了 还重装了好几次系统 快崩溃了都 后来在github上找到的成功案例 总算圆满完成!

  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值