为什么tf.test.is_gpu_available()没有提供true或false,它被卡住了?

问题描述 投票:1回答:1

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

  1. ubuntu 18.04
  2. cuda 10.0
  3. cudnn 7.4.1
  4. 创建了虚拟环境
  5. 安装tensorflow-gpu = 2.0.0

[尝试通过tf.test.is_gpu_available()检查gpu时,卡住了,如下所示。

enter image description here

tensorflow ubuntu-18.04 cudnn
1个回答
0
投票

将cudnn版本更改为7.6.2。然后它很好用。

© www.soinside.com 2019 - 2024. All rights reserved.