google colab 的 GPU 是否适用于 tensorflow 1.0.0？答案

【问题标题】：Does google colab's GPU work with tensorflow 1.0.0?google colab 的 GPU 是否适用于 tensorflow 1.0.0？
【发布时间】：2018-05-30 20:47:35
【问题描述】：

当我在 google colab notebook 中检查 tensorflow 的版本时，我猜它是 1.8.0 预装的。我从笔记本设置中启用了 GPU 并通过此命令进行了检查

import tensorflow as tf
device_name = tf.test.gpu_device_name()
if device_name != '/device:GPU:0':
  raise SystemError('GPU device not found')
print('Found GPU at: {}'.format(device_name))

它给出了想要的输出

但我的代码需要 tensorflow 1.0，所以我使用!pip install tensorflow==1.0 安装它安装成功，但是当我重新启动运行时并检查 GPU 时，它给出了以下错误

系统错误：未找到 GPU 设备

我错过了什么还是 colab 的？

【问题讨论】：

标签： python tensorflow pip gpu google-colaboratory

【解决方案1】：

它可以工作，因为它只是一个 nvidia-docker 容器（或者可能是等效的）。

只有您必须手动安装兼容的 CUDA、cuDNN 和其他软件包。这可能是一个忙碌的过程，我还没有亲自尝试过。此外，您正在安装未启用 gpu 的 tensorflow 包。尝试相同的命令，但使用 tensorflow-gpu 即

!pip install tensorflow-gpu=1.0

【讨论】：

【解决方案2】：

你需要把cuda版本改成8....

这将删除当前版本：

      !apt-get remove cuda
      !apt-get autoremove cuda
      !apt-get purge cuda
      !apt-key del /var/cuda-repo-9-2-local/*.pub
      !rm -rf /var/cuda-repo-8-0-local-ga2/

现在安装版本 8：

     !sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb
     !sudo apt-get update
     !sudo apt-get -y install cuda-8-0

现在安装 tensorflow gpu：

     !pip install tensorflow-gpu==1.0

确保您已安装 tf 1.0 版。

检查它是否有效：

    import tensorflow as tf
    tf.test.gpu_device_name()

【讨论】：