tensorflow报错
本文最后更新于:2024年3月18日 下午
tensorflow报错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.
问题
在运行tensorflow时出现报错信息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.
解决方法
方法一:
在项目中添加
1 |
|
可以正常运行,但此时没有使用GPU,而是使用的CPU进行运算
方法二:
重新安装cuDNN,在官网上下载对应的cuDNN版本,解压,将文件夹内的所有文件复制到CUDA安装目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
下
再次运行可正常使用GPU进行运算
tensorflow报错
http://starnight.top/2020/10/23/tensorflow报错/