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

针对主机上有多个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,paddle,paddledetection如何指定,选择GPU进行训练