如何在Google Colab中获取并升级至TensorFlow 1.7版本?
解决Colab中TensorFlow 1.6升级到1.7的问题
我完全懂你现在的困扰——基于TF 1.7 API写的自定义RNN单元依赖tf.contrib.rnn.LayerRNNCell这个新增属性,但Colab默认的1.6.0版本根本没有它。下面是具体的升级操作步骤,帮你搞定这个问题:
1. 卸载当前TensorFlow版本
在Colab的代码单元格里运行这条命令,先移除现有版本:
!pip uninstall tensorflow -y
2. 安装指定的1.7.0版本
接着安装TensorFlow 1.7.0,一定要明确指定版本号:
!pip install tensorflow==1.7.0
如果需要GPU版本,把命令改成:
!pip install tensorflow-gpu==1.7.0
3. 验证安装并重启运行时
安装完成后,点击Colab菜单栏的「Runtime」→「Restart runtime」重启运行环境,之后运行以下代码确认版本:
import tensorflow as tf print(tf.__version__)
如果输出结果是1.7.0,就说明升级成功了,这时候你的自定义RNN单元就能正常调用LayerRNNCell属性了。
额外小提示
Colab偶尔会自动恢复到默认版本,如果重启后发现版本回退了,重复上面的步骤重新安装即可。
内容的提问来源于stack exchange,提问作者figs_and_nuts
相关产品推荐
相关产品推荐

