admin 管理员组

文章数量: 1087139

如果我的分享对您有用，欢迎点赞，收藏，加关注。创作不易。

针对主机上有多个GPU，一台主机多人使用。不得不指定GPU进行使用。

针对主动指定，选择某一个GPU进行训练的问题。

我再往上搜了很久，没有相关的解决问题。及时相关，也不适用。

我认真查看了一下源码。发现有设置GPU的地方。

我的paddledetection的位置：

 每次用心都会执行这个文件。所以我专门阅读了这个train.py。在train.py的148行，快到文件的末尾了。

 将上面的place = paddle.set_device('gpu')改成place = paddle.set_device('gpu:1')

它默认0号GPU，这里gpu：1意思是设置为1号GPU。主机上的其他GPU不会使用。

这样，一台主机，多个GPU，可以多个工作人员使用。

如果我的分享对您有用，欢迎点赞，收藏，加关注。创作不易。

本文标签： 系统 Windows GPU paddledetection paddle

版权声明：本文标题：windows系统下，多GPU，paddle，paddledetection如何指定，选择GPU进行训练 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://roclinux.cn/b/1726473411a976890.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。