指定GPU
如果要在python代碼中設置使用的GPU（如使用pycharm進行調試時），可以使用下面的代碼
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
制定顯存
定量設置顯存
默認tensorflow是使用GPU盡可能多的顯存。可以通過下面的方式，來設置使用的GPU顯存：
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
按需設置顯存
gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))