问题描述
首先,我在使用和检测GPU(tensorflow-gpu cannot find GPU)时遇到问题。然后我自己找到了解决方案。 Tensorflow搜索名为“ cudnn64_7.dll”的文件。但是cuDNN的版本具有文件名“ cudnn64_8.dll”。所以我重命名了文件。
tf.test.is_built_with_cuda()
tf.test.is_gpu_available(cuda_only=False,min_cuda_compute_capability=None)
tf.test.is_built_with_gpu_support()
重命名文件后,所有代码行均返回true。 Tensorflow最终检测到我的GPU。但是还存在另一个问题。
tensorflow.python.framework.errors_impl.UnkNownError: Failed to get a convolution algorithm. This is probably because cuDNN Failed to initialize,so try looking to see if a warning log message was printed above.
[[node sequential/vgg16/block1_conv1/Relu (defined at /Users/Lenovo/PycharmProjects/AI/test.py:64) ]] [Op:__inference_train_function_1725]
是由文件名引起的吗?如果不是,我该怎么解决?最重要的是哪个版本的cuDNN具有“ cudnn64_7.dll”。我找不到包含该文件的版本。
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)