怎样指定GPU运行程序

连服务器有时候会出现out of memory

解决方法:

换用空闲的gpu

1. 查看GPU


nvidia-smi -L
2. 查看7号GPU


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

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

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

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

则运行命令修改为:


CUDA_VISIBLE_DEVICES=5,6,7 python train.py
 

你可能感兴趣的:(怎样指定GPU运行程序)