指定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显存设置