1、限定TensorFlow使用的GPU
即使目前的任务并不需要很多的GPU资源,TensorFlow在默认情况下任然会占用机器上所有的GPU资源,因此在使用TensorFlow之前,有必要对其使用的GPU资源进行限定。
例如,使用python执行TensorFlow程序,在启动python之前加入如下语句,既可以限定使用GPU编号。
CUDA_VISIBLE_DEVICES=1
CUDA_VISIBLE_DEVICES=”0,1”
CUDA_VISIBLE_DEVICES=0,2,3