Python 谷歌colab'；s GPU与tensorflow 1.0.0配合使用？

当我在GoogleColab笔记本中查看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的？

如果它只是一个nvidia docker容器（或者可能是类似的容器），它会工作的

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

!pip install tensorflow-gpu=1.0

---

考虑到它只是一个nvidia docker容器（或者其他类似的容器），它就可以工作了

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

!pip install tensorflow-gpu=1.0

---

您需要将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()

---

您需要将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()