用户3198739 2022-10-11 21:09 采纳率: 60%
浏览 53
已结题

yolov5模型训练问题

在训练刚开始至图片位置时停止卡了快一个钟了也不动,用cpu训练秒开

img

环境为cuda11.8
cudatoolkit 11.6
python 3.7
pytorch 1.7.0+cu101
torchvision 0.8.1+cu101
gpu是rtx 3070ti laptop

  • 写回答

3条回答 默认 最新

  • 爱晚乏客游 2022-10-12 17:01
    关注

    你的pytorch的cuda版本选错了,torch1.7.0是支持cuda11.x的,但是你下载的是cu101,这个就表示支持的cuda是10.1。但是你环境里面的cuda是11.x(30系显卡不支持cuda11.0以下的版本),所以运行不了。你需要换个cuda11.0的torch就行。另外,torchvision 0.8.1+cu101这个是torch1.7.1的,不匹配会报错的。
    详细的信息可以看

    img


    如果pip安装失败,可以去pip后面的那个网址,使用迅雷下载到本地用pip安装wheel包。另外,我不是很建议将cuda安装在conda的虚拟环境中,最好也是迅雷下载单独安装,然后将cudnn复制到cuda路径下面即可通用。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(2条)

报告相同问题?

问题事件

  • 系统已结题 10月20日
  • 已采纳回答 10月12日
  • 修改了问题 10月11日
  • 修改了问题 10月11日
  • 展开全部