在tensorflow中,我们可以使用 设置使用GPU使用 tf.device 指定Session在第二块GPU上运行:ConfigProto() 中参数 如果安装的是GPU版本的tensorflow,机器上有支持的GPU,也正确安装了显卡驱动、CUDA和cuDNN,默认情况下,Session会在GPU上运行:默认在GPU:0上执行:设置使用cputensorflow中不同的GPU使用/gpu:0和/gpu:1区分,而CPU不区分设备号,统一使用 /cpu:0

详解tf.device()指定tensorflow运行的GPU或CPU设备实现

详解tf.device()指定tensorflow运行的GPU或CPU设备实现