tensorflow - cuDNN 产生错误未能初始化

Question

我正在尝试VGGNET在自定义数据集上训练模型，我已经对模型进行了试验，colab但现在我需要在本地机器上运行相同的代码。

我尝试使用Tensorflow GPU 1.15and运行代码CUDA 10.0（我正在使用的 colab has Tensorflow GPU 1.15and CUDA 10.0），但是代码给出了这个错误

...
  (0) Unknown: 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.
     [[{{node block1_conv1/convolution}}]]
     [[loss/mul/_169]]
...

我尝试过的事情

• 尝试了不同版本的 cuDNN 7.6.4 和 7.4.2
• 尝试将 TensorFlow GPU 从 1.15 降级到 1.14

我也浏览过这些帖子post1，post2但我无法解决错误

系统：Ubuntu 18.04, CUDA 10.0, cuDNN 7.4.2, Nvidia driver 440.64,Tensorflow GPU 1.15

标签： tensorflowgoogle-colaboratoryubuntu-18.04

Answer 1

似乎 RTX 卡存在一些问题，我不得不添加几行代码才能使其工作

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
tf.Session(config=config)