Tensorflow使用不可见的GPU

时间:2019-03-20 13:01:38

标签: python deep-learning gpu

我正在尝试在我可以访问的一部分GPU上训练神经网络。 问题是我只想使用两个GPU,所以我在.bashrc中编写了以下命令:

export CUDA_VISIBLE_DEVICES=6,7

但是,当我观看GPU时,我发现我的程序主要使用内存利用率为100%的其他GPU(4,5),并且仅使用内存利用率很低(最大为9%)的可见GPU。

我已经关闭并重新打开所有终端以重新启动〜/ .bashrc,但问题仍然存在!

感谢您的帮助!

P.S:我之前已经成功地在两个GPU上但在不同的服务器上成功运行了相同的程序,因此内存在这里并不是一个约束。

0 个答案:

没有答案