返回信息流要用pytorch语句选择cuda进行运行
这是一条镜像帖。来源：北邮人论坛 / iwhisper / #7581782同步于 2024/9/13
该镜像源已超过 30 天没有更新，可能在源站已被删除。
IWhisper机器人发帖
为啥yolo运行时只使用cpu在跑
zlql
2024/9/13镜像同步7 回复
订阅后，新回复会通过你的通知中心匿名送达。
7 条回复
rt，装的是cuda11.8，pytorch2.1.1，gpu版的， NVIDIA显卡驱动版本跟cuda也是对应的，但运行yolov5时是用cpu跑的，根本没调用gpu<img src="/img/ubb/ema/1.gif" alt="ema1" style="display:inline;border-style:none">
我是将cuda文件夹通过目录挂载到docker里，然后在docker里运行yolo的，不知道这样做有啥要注意的吗<br>【 在 IWhisper#295 的大作中提到: 】<br><font class="f006">: 要改参数cpu为0或者1吧，你在参数文件里面找找 </font><br>:
选了的，之前还能跑通，但整套迁移到新服务器上就变成用cpu在跑了，正在考虑是不是python版本不对<br>【 在 IWhisper#759 的大作中提到: 】<br><font class="f006">: 和我前几天一个毛病，选择CUDA还要看好device，还要转移上去 </font><br>: