tensorflow - GPU 加速
首先检测是否可用 GPU import tensorflow as tf print('GPU', tf.test.is_gpu_available()) # GPU True tf.device Tensorflow 通过 tf.device 指定每个操作运行的设备,可以指定本地的 CPU、GPU,还可以指定远程服务器; Tensorflow 会给每个本地设备一个名称,如 /cpu:0,即使电脑有多块 CPU ,tf 不会做区分,统一叫 /cpu:0,而 如果有多块 GPU,第 n 块 GPU 叫 /gpu:n,n 从 0 开始; with tf.device('/cpu:0'): d1 = tf.Variable(1.) d2 = tf.Variable(2., name='d2') with tf.device('/gpu:0'): d3 = tf.add(d1, d2) with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as sess: sess.run(tf.global_variables_initializer()) print(sess.run(d3)) log_device_placement tf 提供了 log_device_placement 来查看 计算在 哪个设备上运行;