python-3.x - Google Colab GPU 环境搭建

Question

我正在尝试在 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.xtensorflowgpugoogle-colaboratory

Answer 1

此链接中提供了 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