Issue
Am using Tensorflow 2.0 on Ubuntu 18.04. On running
tf.config.list_physical_devices('GPU')
I get the above error. What is the workaround for this?
Solution
Run tf.config.experimental.list_physical_devices('GPU')
to see all the GPUs
or
run tf.config.experimental.list_physical_devices(device_type=None)
to see all the devices
You can also use tf.test.is_gpu_available()
. It says True if it detects the available gpu.
Answered By - Malgo
0 comments:
Post a Comment
Note: Only a member of this blog may post a comment.