如何在 google colab 中将 tensorflow 永久升级到 2.0

Question

我正在尝试将 tensorflow 从 google colab 中的当前现有版本升级到 2.x，但每次我打开 colab 并检查其版本时，它都会回滚到以前的版本 (1.15.0)。 如何将 tensorflow 永久升级到 2.x？ 因为每次安装都会占用大量内存。 每次运行笔记本时，我一直在尝试命令 !pip install tensorflow==2.1.0，因为它正在回滚到以前的版本。

Answer 1

Colab 预装了两个版本的 TensorFlow：1.x 版本和 2.x 版本。 Colab 目前默认使用 TensorFlow 1.x。

运行import tensorflow将导入默认版本（当前为 1.x）。 在运行 import tensorflow之前，您可以通过使用tensorflow_version魔法运行单元格来使用tensorflow_version

%tensorflow_version 2.x

输出：

TensorFlow 2.x selected.

一旦你通过这个魔法指定了一个版本，你可以像import tensorflow一样运行import tensorflow并验证哪个版本被导入，如下所示：

import tensorflow
print(tensorflow.__version__)

输出：

2.1.0

如果要在导入后切换 TensorFlow 版本，则需要使用“运行时”->“重新启动运行时...”重新启动运行时，然后在再次导入之前指定版本。

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

输出：

1.15.0