could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

前端 未结 19 2505
故里飘歌
故里飘歌 2020-12-01 16:01

I installed tensorflow 1.0.1 GPU version on my Macbook Pro with GeForce GT 750M. Also installed CUDA 8.0.71 and cuDNN 5.1. I am running a tf code that works fine with non C

19条回答
  •  执笔经年
    2020-12-01 16:32

    I also get same error, and I resolved the issue. My system properties were as follows:

    • Operating System: Ubuntu 14.04
    • GPU: GTX 1050Ti
    • Nvidia Driver: 375.66
    • Tensorflow: 1.3.0
    • Cudnn: 6.0.21 (cudnn-8.0-linux-x64-v6.0.deb)
    • Cuda: 8.0.61
    • Keras: 2.0.8

    How I solved the issue is as follows:

    1. I copied cudnn files to appropriate locations (/usr/local/cuda/include and /usr/local/cuda/lib64)
    2. I set the environment variables as:

      * export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"
      * export CUDA_HOME=/usr/local/cuda
      
    3. I also run sudo ldconfig -v command to cache the shared libraries for run time linker.

    I hope those steps will also help someone who is about to go crazy.

提交回复
热议问题