saas222 2022-11-22 19:53 采纳率: 54%
浏览 20
已结题

yolo v5如何运用cuda去训练模型

yolo v5,请问如图所示,我的电脑明明是有cuda,为什么在pycharm里却显示不可用呢?那我应该如何使用cuda去训练模型呢?

img

img

  • 写回答

1条回答 默认 最新

  • PellyKoo 2022-11-22 20:42
    关注

    第一张图只能证明你安装了nvidia驱动,但可能未安装cuda,
    如果已经安装cuda,就要安装torch的cuda版本,而不是cpu版本。
    安装cuda版的torch命令应该是

    pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
    
    

    img

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 系统已结题 12月4日
  • 已采纳回答 11月26日
  • 创建了问题 11月22日