admin管理员组

文章数量:1531721

方法1

基本上这个就能解决了！！！
在train.py的最上方加上下面这两行，注意是最上面，其次指定的就是你要使用的显卡

import os
os.environ['CUDA_VISIBLE_DEVICES']='6'

方法2：

**问题：**命令行参数指定的是4号卡，但实际却总是在0号卡建立进程
真抽象啊，这一步，模型被送到0号卡，但实际上，送到了4号卡（进程是在4号卡上建立的）
解决办法：
在train.py的train()函数下，第一行选择显卡：

device=torch.device('cuda:4')

自己指定显卡即可，也可以在命令行添加一遍，不冲突，但要一样的

基本上方法一就能解决了。

本文标签： 显卡Yolo

版权声明：本文标题：解决yolo无法指定显卡的问题，实测v5、v7、v8有效 内容由热心网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：https://m.elefans.com/dongtai/1726390797a1068571.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。