Colab's TensorFlow 2 возвращается к Tensorflow 1 после перезапуска времени выполнения

Question

Я перезапустил TF с 1.15 до TF2, используя команду:

!pip install tensorflow==2.0
import tensorflow as tf
print(tf.__version__)

Однако, когда я перезагружаю свой ноутбук colab (работает над ячейкой, комментируя первую строку),

Это говоритмоя версия TF 1.15?

Что я должен сделать, чтобы тензорный поток остался в версии 2?

Спасибо,

CS

Answer 1

Поместите команду установки ! pip install tensorflow==2.0 в свою собственную ячейку кода, а импорт - в отдельную ячейку после ячейки установки.

Если вы хотите сохранить установку, сохраните ее на диске Google и выполнитеследующее:После запуска ! pip install tensorflow==2.0 в ячейке под ним запустите;

from google.colab import drive
drive.mount('/content/drive')
! pip freeze --local > /content/drive/My\ Drive/installed.txt

#`then in another cell run` (this is what you'll need to run always to restore your installations)

from google.colab import drive
drive.mount('/content/drive')
! pip install --upgrade --force-reinstall `cat/content/drive/My\ Drive/installed

Затем перейдите по ссылке, чтобы получить токен доступа для подключения вашего диска.

Answer 2

Волшебная команда также является опцией:

try:
  # Use the %tensorflow_version magic if in colab.
  %tensorflow_version 2.x
except Exception:
  pass

import tensorflow as tf

Найден код из примера Udacity: https://colab.research.google.com/github/tensorflow/examples/blob/master/courses/udacity_intro_to_tensorflow_for_deep_learning/l02c01_celsius_to_fahrenheit.ipynb#scrollTo=-ZMgCvSRFqxE