如何在谷歌协作平台上利用GPU升级Tensorflow

Question

目前Google Colaboratory使用的是TensorFlow 1.4.1版本，我想将其升级到1.5.0版本。每次执行!pip install --upgrade tensorflow命令时，笔记本实例都会成功升级TensorFlow版本到1.5.0。但是，在升级操作后，TensorFlow实例只支持“CPU”。

当我执行这个命令时，它什么也没有显示： from tensorflow.python.client import device_lib device_lib.list_local_devices() 是否有其他升级TensorFlow的方法？例如升级到tensorflow-gpu包？此外，笔记本何时将配备升级后的TensorFlow？

- AGP

Answer 1

截至2020年，Colab可以运行Tensorflow 2.0。请卸载当前版本的Tensorflow：

!pip uninstall tensorflow

只需使用pip进行重新安装即可。

!pip install tensorflow==2.0.0

如果您愿意，您还可以安装测试版本。

!pip install tensorflow==2.0.0-beta1

如果您重新安装了Tensorflow，请确保在“运行时”选项卡下重新启动运行时。

一个本地解决方案是以以下方式启动您的代码：

%tensorflow_version 2.x

- DickyBrown

---

我已经尝试过重新安装，但在重新启动运行时并导入tensorflow后，它并不是我尝试重新安装的2.0.0版本。 - AleB

据我所知，Colab有一个版本的tensorflow，它会不时地更新，在2.x这个总称下。如果你想要一个特定的版本，我不确定是否有解决方案，但我很感谢其他用户的意见。 - DickyBrown

Answer 2

即使您安装了GPU版本的!pip install tensorflow-gpu==1.5.0，由于cuda库的原因，它仍然会在导入时失败。目前我还没有找到使用GPU的1.5版本的方法。因此，与其使用没有GPU支持的1.5版本，我宁愿使用带有GPU支持的1.4.1版本。

您可以发送反馈(主页-发送反馈)，希望如果有足够多的人发送类似的内容，他们将更新新的GPU版本。

- Salvador Dali

---

我对此有所怀疑。感谢你的回答。 - AGP

Answer 3

谷歌协作平台似乎支持最新的tensorflow版本，包括1.6.0rc1。

import tensorflow as tf
tf.__version__

#'1.6.0-rc1'

如果你启用GPU硬件加速器，那么GPU应该可以工作。然后，你可以测试一下是否正常运行：

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

输出：

[name: "/device:CPU:0"
 device_type: "CPU"
 memory_limit: 268435456
 locality {
 }
 incarnation: 14621691266205111434, name: "/device:GPU:0"
 device_type: "GPU"
 memory_limit: 198836224
 locality {
   bus_id: 1
 }
 incarnation: 17821632640358169265
 physical_device_desc: "device: 0, name: Tesla K80, pci bus id: 0000:00:04.0, compute capability: 3.7"]

- weiji14

---

是的，我也确认了，谢谢。给反馈只是为了完成工作 :) - AGP

不需要执行“！pip install -U tensorflow-gpu”。如果启用了GPU加速，它就可以正常工作。 - AGP

我明白了，已经编辑过了，移除了pip install tensorflow-gpu :) - weiji14

降级到 v1.4.1，因为我似乎无法使用 v1.6.0 加载预训练权重。还有其他人遇到这个问题吗？请参见：https://github.com/tensorflow/tensorflow/issues/17269 - michael

Answer 4

编辑：这个方法行不通。请查看Salvador的答案。

卸载tensorflow

!pip uninstall tensorflow -y

安装 tensorflow-gpu。

!pip install tensorflow-gpu==1.5.0

- Shoeboom

---

我会尝试给你反馈。但是CUDA 9和cudnn 7怎么办？TensorFlow 1.5.0需要这些软件包。 - AGP