Tensorflow Keras说我的cuda驱动程序是不兼容的,但实际上它们是兼容的版本

Tensorflow Keras说我的cuda驱动程序是不兼容的,但实际上它们是兼容的版本,tensorflow,keras,Tensorflow,Keras,我运行了一个简单的keras脚本,在MNIST数据库上训练conv-net。此脚本可以在我的笔记本电脑上运行,但不能在带有GeForce RTX 2070图形卡的PC上运行 错误是: File "/home/squall/spencer/kaggle/understanding_cloud_organization/mnist_model.py", line 67, in <module> validation_data=(x_test, y_test)) File

我运行了一个简单的keras脚本,在MNIST数据库上训练conv-net。此脚本可以在我的笔记本电脑上运行,但不能在带有GeForce RTX 2070图形卡的PC上运行

错误是:

  File "/home/squall/spencer/kaggle/understanding_cloud_organization/mnist_model.py", line 67, in <module>
    validation_data=(x_test, y_test))
  File "/home/squall/anaconda3/envs/thunder/lib/python3.6/site-packages/keras/engine/training.py", line 1239, in fit
    validation_freq=validation_freq)
  File "/home/squall/anaconda3/envs/thunder/lib/python3.6/site-packages/keras/engine/training_arrays.py", line 196, in fit_loop
    outs = fit_function(ins_batch)
  File "/home/squall/anaconda3/envs/thunder/lib/python3.6/site-packages/tensorflow/python/keras/backend.py", line 3292, in __call__
    run_metadata=self.run_metadata)
  File "/home/squall/anaconda3/envs/thunder/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1458, in __call__
    run_metadata_ptr)
tensorflow.python.framework.errors_impl.UnknownError: 2 root error(s) found.
  (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 conv2d_1/convolution}}]]
     [[metrics/accuracy/Identity/_91]]
  (1) 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 conv2d_1/convolution}}]]
0 successful operations.
0 derived errors ignored.
文件“/home/squall/spencer/kaggle/understanding_cloud_organization/mnist_model.py”,第67行,在
验证数据=(x_检验,y_检验)
文件“/home/squall/anaconda3/envs/thunder/lib/python3.6/site-packages/keras/engine/training.py”,第1239行,适合
验证频率=验证频率)
文件“/home/squall/anaconda3/envs/thunder/lib/python3.6/site packages/keras/engine/training_arrays.py”,第196行,在fit_循环中
outs=配合功能(ins\U批量)
文件“/home/squall/anaconda3/envs/thunder/lib/python3.6/site packages/tensorflow/python/keras/backend.py”,第3292行,在调用中__
run\u元数据=self.run\u元数据)
文件“/home/squall/anaconda3/envs/thunder/lib/python3.6/site packages/tensorflow/python/client/session.py”,第1458行,在__
运行_元数据_ptr)
tensorflow.python.framework.errors\u impl.UnknownError:找到2个根错误。
(0)未知:获取卷积算法失败。这可能是因为cuDNN未能初始化,所以请尝试查看上面是否打印了警告日志消息。
[{{node conv2d_1/卷积}]]
[[metrics/Accurance/Identity/_91]]
(1) 未知:获取卷积算法失败。这可能是因为cuDNN未能初始化,所以请尝试查看上面是否打印了警告日志消息。
[{{node conv2d_1/卷积}]]
0成功的操作。
忽略0个派生错误。
Cuda是10.1。司机是418.56。CuDNN是7.4.2。Tensorflow为1.14。根据Nvidia官方图表,这些都是兼容版本

有什么想法吗?

试试这个


PS:CUDA是10.0,cuDNN是7.6.3。对于CUDA10.0,您需要CUDA 10版本。完全卸载Cuda 10.1并安装受支持的Cuda 10。您可以阅读TF的要求。

两个版本不匹配,请查看Nvidia网站