Tensorflow仅显示“在本地成功打开了CUDA库libcublas.so.10.0”，而关于cudnn则没有任何显示

Question

我的张量流只打印出这一行：

I tensorflow/stream_executor/dso_loader.cc:152] successfully opened CUDA library libcublas.so.10.0 locally 运行时。

网上的Tensorflow日志有很多其他库正在加载，例如libcudnn。

因为我认为我的安装性能不是最佳，所以我试图找出是否是因为这个原因。任何帮助将不胜感激！

我的tf是1.13.1 NVIDIA驱动程序版本：418.67 CUDA版本：10.1（我还安装了10.0。这可能是问题吗？）

Answer 1

根据TensorFlow 文档，cuDNN是 的要求tensorflow-gpu。如果没有cuDNN安装，则无法安装tensorflow-gpu因为依赖库会丢失。

因此，如果您已成功安装tensorflow-gpu并能够使用它，例如

import tensorflow as tf
tf.Session()

你很好。

编辑

我只是在这里检查，tensorflow_gpu-1.13.1官方只支持CUDA 10.0. 我建议使用它而不是CUDA 10.1.

此外，NVIDIA建议使用410.48带有CUDA 10.0. 我也会坚持下去。