(阅读时间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'