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

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 wincc已组态的变量过多
  • ¥60 如图:直线与椭圆X轴平行,求直线与椭圆任意一点的相切坐标计算公式
  • ¥50 如何用python使用opencv里的cv::cudacodec::VideoWriter函数对视频进行GPU硬编码
  • ¥100 c#solidworks 二次开发 工程图自动标边线法兰 等折弯尺寸怎么标
  • ¥15 halcon DrawRegion 提示错误
  • ¥15 FastAPI Uvicorn启动显示404
  • ¥15 centos7.9脚本,怎么排除特定的访问记录
  • ¥15 关于#Django#的问题:我的静态文件呢?
  • ¥15 关于CPLEX的问题,请专家解答
  • ¥15 cocos的点击事件 怎么穿透到 原生fragment上。