如何讓 cudnn 工作？ （初始化失敗）

Question

在 Windows 10 上出現此錯誤

UnknownError: 獲取卷積算法失敗。 這可能是因為 cuDNN 初始化失敗，所以嘗試查看上面是否打印了警告日志消息。 [[{{節點 conv2d_1/卷積}} = Conv2D[T=DT_FLOAT, _class=["loc:@training_1/Adam/gradients/conv2d_1/convolution_grad/Conv2DBackpropFilter"], data_format="NCHW", dilations=[1, 1 , 1, 1], padding="有效", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:GPU:0" ](training_1/Adam/gradients/conv2d_1/convolution_grad/Conv2DBackpropFilter-0-TransposeNHWCToNCHW-LayoutOptimizer, conv2d_1/kernel/read)]] [[{{node loss_1/mul/_267}} = _Recvclient_terminated=false, recv_device="/job :localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_782_loss_1/ mul", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"]]

我有 RTx 2070 並且：

Python 3.6.5 tf 1.12.0 tf-gpu 1.12.0 cuda 9.0 所有補丁。 cudnn 7.3.1 凱拉斯 2.2.4

我知道 cudnn 的 nvdia 頁面，我在這里閱讀了一些其他答案。 我對遺漏的小細節很感興趣。 將CUDA文件夾中的3個文件移動到3個目錄后，是不是還要多一步？ 也許有不同部件需要安裝的順序？

Cuda 似乎工作正常，python 看到了，matlab 也看到了。

在為我從網上獲得的 mnist 運行此代碼時發生錯誤，如果我卸載 tensorflow-gpu 並在 cpu 上使用 tensorflow，則該代碼有效。

過去有很大幫助的一個例子是，除非您自定義並取消選中 visual studio 選項，否則您無法安裝 cuda。 謝謝 ！

Answer 1

使用 CUDA 10 的 RTX 2070 卡也有類似的問題......

解決方案是使用：

config.gpu_options.allow_growth = True

在張量流中。

有關如何使用該參數的更多信息： How to prevent tensorflow from allocating the totality of a GPU memory?

Answer 2

我認為 RTX 卡需要 CUDA 10 。