1. 程式人生 > >tensorflow_gpu InvalidArgumentError: Cannot assign a device for operation 'add' 錯誤

tensorflow_gpu InvalidArgumentError: Cannot assign a device for operation 'add' 錯誤

在新安裝的tensorflow gpu 版本上編譯程式碼,之前程式碼在cpu 上面都是可以執行的。

報如下錯誤:

InvalidArgumentError: Cannot assign a device for operation 'add': Operation was explicitly assigned to /device:GPU:1 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:GPU:0 ]. Make sure the device specification refers to a valid device.      [[{{node add}} = Add[T=DT_FLOAT, _device="/device:GPU:1"](a_1, b_1)]]

解決方法:

將:

sess = tf.Session()

改為:

sess = tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True))