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

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 想通过pywinauto自动电机应用程序按钮，但是找不到应用程序按钮信息
  • ¥15 MATLAB中streamslice问题
  • ¥15 如何在炒股软件中，爬到我想看的日k线
  • ¥15 51单片机中C语言怎么做到下面类似的功能的函数（相关搜索：c语言）
  • ¥15 seatunnel 怎么配置Elasticsearch
  • ¥15 PSCAD安装问题 ERROR: Visual Studio 2013, 2015, 2017 or 2019 is not found in the system.
  • ¥15 (标签-MATLAB|关键词-多址)
  • ¥15 关于#MATLAB#的问题，如何解决？（相关搜索：信噪比，系统容量）
  • ¥500 52810做蓝牙接受端
  • ¥15 基于PLC的三轴机械手程序