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

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日

悬赏问题

  • ¥15 metadata提取的PDF元数据，如何转换为一个Excel
  • ¥15 关于arduino编程toCharArray()函数的使用
  • ¥100 vc++混合CEF采用CLR方式编译报错
  • ¥15 coze 的插件输入飞书多维表格 app_token 后一直显示错误，如何解决？
  • ¥15 vite+vue3+plyr播放本地public文件夹下视频无法加载
  • ¥15 c#逐行读取txt文本，但是每一行里面数据之间空格数量不同
  • ¥50 如何openEuler 22.03上安装配置drbd
  • ¥20 ING91680C BLE5.3 芯片怎么实现串口收发数据
  • ¥15 无线连接树莓派，无法执行update，如何解决？（相关搜索：软件下载）
  • ¥15 Windows11, backspace, enter, space键失灵