- 一般可能是因为cudnn的版本没有安装对,先用下边命令来检测一下,看一下你cudnn的版本对应的对不对?
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 #此命令适用于Ubuntu系统,如果是Windows,那么百度搜一下怎么检查cudnn版本
- 你的显存暴炸了,也就是GPU没有分配好,在你使用的代码中添加下边的代码就好了。
gpus = tf.config.experimental.list_physical_devices(device_type='GPU') tf.config.experimental.set_memory_growth(gpus[0],True)
- 看看你的代码是不是有问题,tensorflow1.0写的代码,不能直接在tensorflow2.0以上版本运行
Unknown: Failed to get convolution algorithm.
最新推荐文章于 2021-11-02 14:22:03 发布