E tensorflow / stream_executor / cuda / cuda_dnn.cc:329]无法创建cudnn句柄:CUDNN_STATUS_ALLOC_FAILED

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

系统信息

OS平台和发行版:Windows 10版本1909TensorFlow GPU版本:2.0.0python版本:3.6.10CUDA版本:10.0.130cuDNN版本:7.6.5Nvidia Graphics版本:441.87GPU型号和内存:英特尔(R)UHD图形630,共享GPU内存:15.9GBNvidia GeForce GTX 1660 Ti,专用GPU内存:6GB内存:32GB(31.8GB)可用:25.6GB

这是我的gpu详细信息:

2020-01-10 09:40:58.352755: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1618] Found device 0 with properties:
name: GeForce GTX 1660 Ti major: 7 minor: 5 memoryClockRate(GHz): 1.455 pciBusID: 0000:01:00.0
2020-01-10 09:40:58.356851: I tensorflow/stream_executor/platform/default/dlopen_checker_stub.cc:25] GPU libraries are statically linked, skip dlopen check.
2020-01-10 09:40:58.359863: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1746] Adding visible gpu devices: 0
2020-01-10 09:40:58.361866: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
2020-01-10 09:40:58.367388: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1618] Found device 0 with properties:
name: GeForce GTX 1660 Ti major: 7 minor: 5 memoryClockRate(GHz): 1.455 pciBusID: 0000:01:00.0
2020-01-10 09:40:58.372233: I tensorflow/stream_executor/platform/default/dlopen_checker_stub.cc:25] GPU libraries are statically linked, skip dlopen check.
2020-01-10 09:40:58.375350: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1746] Adding visible gpu devices: 0
2020-01-10 09:40:58.940556: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1159] Device interconnect StreamExecutor with strength 1 edge matrix:
2020-01-10 09:40:58.944788: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1165]      0
2020-01-10 09:40:58.946605: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1178] 0:   N
2020-01-10 09:40:58.948791: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1304] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 4629 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1660 Ti, pci bus id: 0000:01:00.0, compute capability: 7.5)  

此后,我收到错误消息:

2020-01-10 09:41:01.248321: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED
2020-01-10 09:41:01.252612: E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED  

最后我得到

File "<string>", line 3, in raise_from tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [Op:Conv2D]

这个问题并非毫无价值。这需要一些研究。从那时起,我一直在处理和浏览它,如果您知道请发布您的答案,可悲的是找不到任何解决方案。谢谢您的贡献。

python-3.6 tensorflow2.0 cudnn
1个回答
0
投票

我刚刚更新了几个软件包,然后就可以使用了。

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