1

安装 tensorflow gpu = 2.0.0 后,检测到 gpu 后卡住了。该项目的环境设置是

  1. Ubuntu 18.04
  2. 库达 10.0
  3. cudnn 7.4.1
  4. 创建了一个虚拟环境
  5. 安装 tensorflow-gpu=2.0.0

尝试使用 tf.test.is_gpu_available().compliation 检查 gpu 时卡住了,如下所示。

在此处输入图像描述

4

1 回答 1

0

将 cudnn 版本更改为 7.6.2。然后它运行良好。

于 2020-04-20T13:05:30.473 回答