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]
这个问题并非毫无价值。这需要一些研究。从那时起,我一直在处理和浏览它,如果您知道请发布您的答案,可悲的是找不到任何解决方案。谢谢您的贡献。
我刚刚更新了几个软件包,然后就可以使用了。