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

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日

悬赏问题

  • ¥30 这是哪个作者做的宝宝起名网站
  • ¥60 版本过低apk如何修改可以兼容新的安卓系统
  • ¥25 由IPR导致的DRIVER_POWER_STATE_FAILURE蓝屏
  • ¥50 有数据，怎么建立模型求影响全要素生产率的因素
  • ¥50 有数据，怎么用matlab求全要素生产率
  • ¥15 TI的insta-spin例程
  • ¥15 完成下列问题完成下列问题
  • ¥15 C#算法问题, 不知道怎么处理这个数据的转换
  • ¥15 YoloV5 第三方库的版本对照问题
  • ¥15 请完成下列相关问题！