指定GPU训练 CUDA_VISIBLE_DEVICES

 

方法一

import os

import tensorflow as tf

os.environ["CUDA_VISIBLE_DEVICES"]="-1"    CPU模式

CUDA_VISIBLE_DEVICES=1    Only device 1 will be seen
CUDA_VISIBLE_DEVICES=0,1    Devices 0 and 1 will be visible
CUDA_VISIBLE_DEVICES="0,1"    Same as above, quotation marks are optional
CUDA_VISIBLE_DEVICES=0,2,3    Devices 0, 2, 3 will be visible; device 1 is masked

方法二

CUDA_VISIBLE_DEVICES=1 python **.py

 

方法3

使用 tf.device('/gpu:1') 指定Session在第二块GPU上运行

tensorflow中不同的GPU使用/gpu:0和/gpu:1区分，而CPU不区分设备号，统一使用 /cpu:0