Hi,
I've been using a Deep Learning VM image, and I noticed that I don't have GPU devices loaded in my Jupyter Lab.
When I use:
K.get_session().list_devices()
I get only the following:
[_DeviceAttributes(/job:localhost/replica:0/task:0/device:CPU:0, CPU, 268435456)]
And when I try to run the Tensorflow classical GPU test
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
I get the following:
InvalidArgumentError (see above for traceback): Cannot assign a device for operation 'MatMul': Operation was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0 ]. Make sure the device specification refers to a valid device.
[[Node: MatMul = MatMul[T=DT_FLOAT, transpose_a=false, transpose_b=false, _device="/device:GPU:0"](a, b)]]
How can I make sure that I'm running Tensorflow in a GPU?
Best,
Atilio