GPU device not found in Colab (likely due to most recent TF 2.0 update)

My code worked well with GPU in Colab yesterday. But this morning it became very slow. So I suspect that CPU is used despite hardware accelerator is set to GPU in “change runtime type” explicitly. The following test code result in “SystemError: GPU device not found”.

code chunk:

%tensorflow_version 2.x  
import tensorflow as tf  
device_name = tf.test.gpu_device_name()  
if device_name != '/device:GPU:0':
   raise SystemError('GPU device not found')
print('Found GPU at: {}'.format(device_name))

the same problem has be replicated by other users when I reported the issue to TF team: https://github.com/tensorflow/tensorflow/issues/34385. I posted it here because I think Colab can solve the problem by simply reverse the most recent update to TF 2.0 in their end. thank you.

[jakevdp]

If you are using tensorflow 2.0 on Colab, we recommend using the bundled version, which you can enable with:

%tensorflow_version 2.x

This will switch your tensorflow version to the current 2.X version built for Colab (see https://colab.research.google.com/notebooks/tensorflow_version.ipynb for details)

If you install external releases of tensorflow via pip install tensorflow-gpu==2.0 or similar, it will install a pre-built binary that may not be compatible with the GPUs and drivers available in Colab’s runtime.

[DevashishX]

Is this issue solved now ?

is 2.2.0-rc3 the right version ?