如何指定GPU运行程序

连服务器有时候会出现outof memory

解决方法：

换用空闲的gpu

查看GPU

1.nvidia-smi -L

查看7号GPU

2.nvidia-smi-q -i 7

如果只看memory情况。可以用：

nvidia-smi-q -i 7 -d MEMORY

3.设置GPUid

CUDA_VISIBLE_DEVICES=6(或CUDA_VISIBLE_DEVICES=6,7）command

如：

版权归作者所有，任何形式转载请联系作者。

作者：宝图（来自豆瓣）

来源：https://www.douban.com/note/641416299/

这条命令适用于命令行运行tensorflow程序的时候，指定gpu.

只需要在命令之前设置环境变量，简单来说比如原本程序是命令行运行pythontrain.py

假定这里gpu总共有八块，通过nvidia-smi查看发现5,6,7是空闲的（从0开始编号）

则运行命令修改为：CUDA_VISIBLE_DEVICES=5,6,7python train.py