矩池云中Tensorflow指定GPU及GPU显存设置

指定GPU

查看机器上GPU情况

命令： nvidia-smi

功能：显示机器上gpu的情况

命令： nvidia-smi -l

功能：定时更新显示机器上gpu的情况

其中左侧红框有0、1、2、3的编号，表示GPU的编号，在后面指定GPU时需要使用这个编号。

指定使用哪块GPU

~/ CUDA_VISIBLE_DEVICES=0  python your.py #使用GPU0
~/ CUDA_VISIBLE_DEVICES=0,1 python your.py #使用GPU0,1
# 注意单词不要打错
 
# 或者在程序开头
os.environ['CUDA_VISIBLE_DEVICES'] = '0'  #使用 GPU 0
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'  # 使用 GPU 0，1

限制GPU的使用率

方法一

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.85  #占用85%显存
session = tf.Session(config=config)

方法二

gpu_options=tf.GPUOptions(per_process_gpu_memory_fraction=0.85)
config=tf.ConfigProto(gpu_options=gpu_options)
session = tf.Session(config=config)

参考文章

tf.ConfigProto和tf.GPUOptions用法总结

tensorflow学习笔记（二十五）：ConfigProto&GPU

tensorflow | ConfigProto GPU使用设置（包括占用率等）

tensorflow中指定GPU及GPU显存设置