gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
上面这两个表示是对gpu设置操作,之前都要先import tensorflow as tf,这个是基本操作。这里的设置实际上相当是给每块gpu只占用33%的空间,比如12g的大概只占4g左右。
然后还有一点就是如果想指定多块gpu的话,可以用这个比较简单的方法。
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1,2,3'
这样的话直接就指定了4块gpu,再加上之前写的代码,就可以保证最后的使用。
再提一点就是tf.ConfigProto这个函数就是用来搞gpu的。
网友评论