【发布时间】:2021-01-04 17:28:34
【问题描述】:
安装 tensorflow gpu = 2.0.0 后检测到 gpu 后卡住了。 这个项目的环境设置是
- ubuntu 18.04
- cuda 10.0
- cudnn 7.4.1
- 创建了一个虚拟环境
- 安装 tensorflow-gpu=2.0.0
在尝试使用 tf.test.is_gpu_available().compliation 检查 gpu 时卡住了,如下所示。
【问题讨论】:
-
如果你是第一次启动 Tensorflow?如果是这种情况,请尝试等待 5 到 10 分钟,Tensorflow 需要在您第一次执行程序时加载 CUDA 内核。
-
2020-04-16 16:08:46.499296: I tensorflow/compiler/xla/service/service.cc:175] StreamExecutor 设备 (1): Tesla T4, Compute Capability 7.5 开启后控制台..它卡住了