安装tensorflow gpu = 2.0.0后,检测到gpu后卡住了。此项目的环境设置为
[尝试通过tf.test.is_gpu_available()检查gpu时,卡住了,如下所示。
enter image description here
将cudnn版本更改为7.6.2。然后它很好用。