could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

前端 未结 19 2552
故里飘歌
故里飘歌 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:34

    In Tensorflow 2.0, my issue was resolved by setting the memory growth. ConfigProto is deprecated in TF 2.0, I used tf.config.experimental. My computer specs are:

    • OS: Ubuntu 18.04
    • GPU: GeForce RTX 2070
    • Nvidia Driver: 430.26
    • Tensorflow: 2.0
    • Cudnn: 7.6.2
    • Cuda: 10.0

    The code I used was:

    physical_devices = tf.config.experimental.list_physical_devices('GPU')
    assert len(physical_devices) > 0, "Not enough GPU hardware devices available"
    config = tf.config.experimental.set_memory_growth(physical_devices[0], True)
    

提交回复
热议问题