Google Colab GPU 环境搭建
Posted
技术标签:
【中文标题】Google Colab GPU 环境搭建【英文标题】：Google Colab GPU environment establish 【发布时间】：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的输出
【问题讨论】：
【参考方案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
【讨论】：
以上是关于Google Colab GPU 环境搭建的主要内容，如果未能解决你的问题，请参考以下文章
史上最最最详尽的CentOS 7深度学习GPU环境搭建！收藏干货