Tensorflow 指定GPU运行

参考网址：http://blog.csdn.net/u014381600/article/details/72911262

1.终端执行程序时设置使用的GPU（linux终端）

CUDA_VISIBLE_DEVICES=1,2  python my_script.py

想指定几块GPU就标明GPU序号即可，中间用逗号隔开

2.python代码中设置使用的GPU（ipython或pycharm调试时）

import os 

os.environ["CUDA_VISIBLE_DEVICES"] = "2"

3.设置tensorflow使用的显存大小

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7) 

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))