Tf-gpu调用错误:Failed to get convolution algorithm. This is probably because cuDNN failed to initialize.

版本:tensorflow2.5 + CUDA11.2 + NVIDIA-SMI 472.19

报错代码

UnknownError:  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.
	 [[node model/conv2d/Conv2D (defined at Users\用户名\AppData\Local\Temp\ipykernel_24240\3644285622.py:37) ]] [Op:__inference_train_function_7180]

Function call stack:
train_function


跑深度学习的时候报错,这个问题网上的教程也说得很清楚,要么是CUDA,要么是显存。
CUDA的问题,除了版本确实安装得有问题,还要注意环境变量路径。

在我重新安装好所有版本后,版本都一致了,但是报错,然后检查发现是环境变量路径的问题,当系统内存在多个CUDA版本的时候,最新版本的路径会在旧版本之上,所以要确保自己想要使用的CUDA版本是当前路径。

系统属性-高级-环境变量-系统变量-PATH,看这里的路径,想要使用的CUDA版本是不是在最前面,选完记得逐个确定。

可以在终端用nvcc --version检查系统当前的CUDA是什么版本,像上图当前版本就是11.2。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值