报错:Cannot assign a device for operation 'HorovodAllreduce_random_uniform_0': Could not satisfy explicit device specification '/device:GPU:1' because no supported kernel for GPU devices is available.
解决办法:加上 config = tf.ConfigProto(allow_soft_placement=True) ,默认使用 GPU
import tensorflow as tf from time import sleep a = tf.Variable([1,2,3,4,5]) # 加上下面一行 config = tf.ConfigProto(allow_soft_placement=True) # 设置 gpu 随使用增长 config.gpu_options.allow_growth = True sess = tf.Session(config=config) sess.run(tf.global_variables_initializer()) print(sess.run(a))