无法在Google Colab上使用GPU来运行TensorFlow 2.0。

Question

我正在尝试在Google Colab上使用GPU运行笔记本，但它没有为我提供GPU。然而，当我使用tensorflow 1.15.0运行笔记本时，GPU是可用的。

tf.test.gpu_device_name()

在tensorflow 1.15.0中，该函数输出'/device:GPU:0'

但是，在tensorflow 2.0.0中，相同的函数返回''。

- Alwin Aind

---

你已经在硬件加速中启用了GPU吗？ - thisisjaymehta

@thisisjaymehta 是的，我已经在硬件加速中启用了GPU，并且该函数返回'/device:GPU:0'，适用于tensorflow 1.15.0。 - Alwin Aind

尝试卸载所有旧版tensorflow并安装tensorflow-gpu。 - thisisjaymehta

@thisisjaymehta 仍然是相同的结果。 - Alwin Aind

尝试升级CUDA到最新版本。 - thisisjaymehta

Answer 1

我在Google Colab上成功安装了。

!pip install tensorflow-gpu 

并且

!pip install tf-nightly 

现在运行 tf.test.gpu_device_name()，输出结果是 /device:GPU:0。 但是，TensorFlow自动升级到了版本2.1.0-dev20191120

- Cristina Bustos

Answer 2

尝试。

%tensorflow_version 2.x

import tensorflow as tf
print(tf.__version__)
print(tf.test.gpu_device_name())
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

- Xuanhong Ye

Answer 3

这是一个与Colab最近升级的CUDA 10.1不兼容的2.0相关的错误。

关注相关的GitHub问题: https://github.com/googlecolab/colabtools/issues/864

- Bob Smith

Answer 4

根据https://colab.research.google.com/notebooks/gpu.ipynb#scrollTo=oM_8ELnJq_wd，

您需要为笔记本启用GPU：

1. 导航到“编辑”→“笔记本设置”
2. 从“硬件加速器”下拉菜单中选择GPU

- MerajS