1:在终端执行程序指定GPU
CUDA_VISIBLE_DEVICES=0 python demo.py
其他可用的形式
CUDA_VISIBLE_DEVICES=0,1
CUDA_VISIBLE_DEVICES="0,1,2"
2:在python代码中指定GPU
import os
os.environ["CUDA_VISIBLE_DEVICES"]="0,1"
3:设置定量的GPU使用量
import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.9
session = tf.Session(config=config)
4:设置最小的GPU使用量
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)