解决ubuntu18.04 cuda10.0+cudnn7.5+tensorflow1.13.1安装后出现的问题

    xiaoxiao2022-07-02  113

    本文为博主原创文章,未经博主允许不得转载!

    ubuntu18.04 cuda10.0+cudnn7.5+tensorflow1.13.1 报错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.

    解决办法: 在代码开头导入下面几行代码

    // An highlighted block import tensorflow as tf config = tf.ConfigProto() config.gpu_options.allow_growth = True sess = tf.Session(config=config)
    最新回复(0)