Handling `The default version of TensorFlow in Colab will soon switch to TensorFlow 2.x.`
I started a Colab notebook from scratch and imported TensorFlow, expecting to get version 2. Somehow I get this message:
The default version of TensorFlow in Colab will soon switch to TensorFlow 2.x.We recommend you upgrade now or ensure your notebook will continue to use TensorFlow 1.x via the
%tensorflow_version 1.xmagic: more info.
Too bad it does not automatically switch to TF2 yet. So in order to get the right version, I have to prepend my import with that special flag:
%tensorflow_version 2.x import tensorflow as tf
Upon running the cell again, I get:
TensorFlow is already loaded. Please restart the runtime to change versions.
So I restart the runtime
And run the cell again. This time it loads the right version: