Google Colab GPU 环境搭建

【问题标题】：Google Colab GPU environment establishGoogle Colab GPU 环境搭建
【发布时间】：2021-10-30 02:36:13
【问题描述】：

我正在尝试在 Google Colab 中运行我的深度学习代码，我已经为 tensorflow 1.14.0 安装了 cuda10.0.130 和 cudnn7.6.4，但是 tf.test.is_gpu_available() 的结果仍然是错误的，我不知道是什么我现在可以做吗，有人可以给我一些指示吗？这是!sudo lsb_release -a和!nvidia-smi的输出

【问题讨论】：

标签： python-3.x tensorflow gpu google-colaboratory

【解决方案1】：

link 中提供了 GPU 版本支持和测试的配置

Cuda 10.1 和 Cudnn 7.6 的支持版本将是 tensorflow_gpu-2.3.0
TF 1.X 版本 CPU 和 GPU 支持也不同

所以你应该这样做

!pip install tensorflow_gpu==1.14.0

用于使用 GPU 版本的 Tensorflow

参考-https://www.tensorflow.org/install/gpu#older_versions_of_tensorflow

【讨论】：