Как обновить tenorflow до 2.0 в google colab навсегда - PullRequest
0 голосов
/ 23 марта 2020

Я пытаюсь обновить tenorflow до 2.x с текущей существующей версии в google colab, но каждый раз, когда я открываю colab и проверяю его версию, он возвращается к предыдущей версии (1.15.0). Как надолго обновить тензор потока до 2.х? так как он занимает много памяти для установки каждый раз. Я пробовал команду! Pip install tenorflow == 2.1.0 каждый раз при запуске ноутбука, так как он получает откат к предыдущей версии.

1 Ответ

1 голос
/ 23 марта 2020

В Colab предварительно установлены две версии TensorFlow: версия 1.x и версия 2.x. В настоящее время Colab использует TensorFlow 1.x по умолчанию.

Запуск import tensorflow импортирует версию по умолчанию (в настоящее время 1.x). Вы можете использовать 2.x, запустив ячейку с tensorflow_version волхвами c до того, как запустите импорт тензор потока. '

%tensorflow_version 2.x

Вывод:

TensorFlow 2.x selected.

После того, как вы указали версию с помощью этой магии c, вы можете запустить import tensorflow в обычном режиме и проверить, какая версия была импортирована, следующим образом:

import tensorflow
print(tensorflow.__version__)

Вывод:

2.1.0

Если вы хотите переключать версии TensorFlow после импорта, вам потребуется перезапустить среду выполнения с помощью «Runtime» -> «Restart runtime ...» и затем указать версию, прежде чем импортировать ее снова.

%tensorflow_version 1.x
import tensorflow as tf
print(tf.__version__)

Выход:

1.15.0
...