PyTorch : RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

TensorFlow ImportError: libcublas.so.9.0: cannot open shared object file:

PyTorch : RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED 

这两个问题放在一起解答,而引起的原因不同。

PyTorch : RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED, Cudnn存在的问题。自从Pytorch推出了1.0和1.1之后,开始使用Cudnn来加速RNN。而原生的Python3对此支持不够。

最好使用Conda python来解决这个问题。

 

TensorFlow ImportError: libcublas.so.9.0: cannot open shared object file:

出现这个问题往往是cudnn和cuda版本不够匹配的原因,甚至cudnn没有安装好,参考我写的博客。TensorFlow 1.13 配合cuda 9 或者cuda 10,再配合cudnn 7.4 / 7.5.

TensorFlow 1.12一下,配cuda9 和cuda 10就行了。

你可能感兴趣的:(Python,Cuda)