Unknown: Failed to get convolution algorithm.

  1. 一般可能是因为cudnn的版本没有安装对,先用下边命令来检测一下,看一下你cudnn的版本对应的对不对? 
    cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
    #此命令适用于Ubuntu系统,如果是Windows,那么百度搜一下怎么检查cudnn版本

  2.        你的显存暴炸了,也就是GPU没有分配好,在你使用的代码中添加下边的代码就好了。       
    gpus = tf.config.experimental.list_physical_devices(device_type='GPU')
    tf.config.experimental.set_memory_growth(gpus[0],True)
                                                                                                                                                                         
  3.  看看你的代码是不是有问题,tensorflow1.0写的代码,不能直接在tensorflow2.0以上版本运行
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章