深度学习TensorFlow系列：设置使用特定GPU

（阅读时间20S）

为防止模型占据太大资源，或者与别人抢资源。需设置TensorFlow使用特定某块GPU，非常容易，这里介绍一个最简单的。

import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1'

这里指定模型在1号显卡上跑。

设置一块显卡使用多少：0.5意思，只用50%的显存。

import tensorflow as tf

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5

那么指定某两块的话，如下，使用0，1两块显卡。

os.environ['CUDA_VISIBLE_DEVICES'] = '0, 1'