yolov7的bug，无法指定显卡（程序默认0号卡）
- 开源代码
- 2025-08-17 14:18:02
**问题：**命令行参数指定的是4号卡，但实际却总是在0号卡建立进程 真抽象啊，这一步，模型被送到0号卡，但实际上，送到了4号卡（进程是在4号卡上建立的） 解决办法： 在train.py的train()函数下，第一行选择显卡：
device=torch.device('cuda:4')自己指定显卡即可，也可以在命令行添加一遍，不冲突，但要一样的
yolov7的bug，无法指定显卡（程序默认0号卡）由讯客互联开源代码栏目发布，感谢您对讯客互联的认可，以及对我们原创作品以及文章的青睐，非常欢迎各位朋友分享到个人网站或者朋友圈，但转载请说明文章出处“yolov7的bug，无法指定显卡（程序默认0号卡）”
![[深度学习]卷积神经网络的概念,入门构建(代码实例)](/0pic/pp_03.jpg)
