device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/dev
解决方法:
sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True))
版权声明:本文为jacke121原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。
device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/dev
解决方法:
sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True))