Impossible d'utiliser le GPU sur Google Colab pour tensorflow 2.0
J'essaie d'exécuter mon bloc-notes à l'aide d'un GPU sur Google Colab, mais il ne me fournit pas de GPU, mais lorsque j'exécute le bloc-notes avec tensorflow 1.15.0, le GPU est disponible.
tf.test.gpu_device_name()
donne la sortie '/device:GPU:0' pour tensorflow 1.15.0
Mais quand je fais de même avec tensorflow 2.0.0, la fonction renvoie ''.
Il s'agit d'un bogue lié à l'incompatibilité de la version 2.0 avec une récente mise à niveau de CUDA 10.1 sur Colab.
Le problème GitHub à suivre est le suivant: https://github.com/googlecolab/colabtools/issues/864