(Tensorflow-GPU) import tensorflow ImportError: Could not find 'cudnn64_7.dll'

后端 未结 4 443
情话喂你
情话喂你 2021-02-05 14:39

After created tensorflow environment under anaconda, I installed tensorflow-gpu. Then I was trying to import tensorflow to verify if it\'s correctly installed, but got this erro

4条回答
  •  小鲜肉
    小鲜肉 (楼主)
    2021-02-05 15:19

    Just fould the solution:

    I checked the \tensorflow\python\platform\build_info.py and found:

    msvcp_dll_name = 'msvcp140.dll'
    cudart_dll_name = 'cudart64_90.dll'
    cuda_version_number = '9.0'
    nvcuda_dll_name = 'nvcuda.dll'
    cudnn_dll_name = 'cudnn64_7.dll'
    cudnn_version_number = '7'
    

    It assumes cudnn version is 7. So just need to correct it as:

    cudnn_dll_name = 'cudnn64_6.dll'
    cudnn_version_number = '6'
    

提交回复
热议问题