在之前电脑更新了tensorflow后,出现问题
tensorflow.python.framework.errors_impl.InternalError: Blas SGEMM launch failed : m=4096, n=64, k=64 [Op:Conv2D]
看了一下网上评论说多半是cudnn的问题,但是重装多次cuda也还是这个问题。
最后通过发现是cudnn(8.1.0)和cuda toolkit版本太高,不匹配,导致,
多方踩坑后,解决问题的tensorflow环境配置:
tensorflow2.4.0
cuda11.0 update1
cudnn8.0.5
另外,安装完cuda后别忘了更新环境变量,以下内容为转载
https://blog.youkuaiyun.com/qq_42167046/article/details/113246994
输入vi ~/.bashrc命令打开文件,在文件结尾输入以下语句,保存。(Linux ssh修改文件)
export PATH=/usr/local/cuda/bin:
P
A
T
H
e
x
p
o
r
t
L
D
L
I
B
R
A
R
Y
P
A
T
H
=
/
u
s
r
/
l
o
c
a
l
/
c
u
d
a
/
l
i
b
64
:
PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:
PATHexportLDLIBRARYPATH=/usr/local/cuda/lib64:LD_LIBRARY_PATH
更新环境变量配置:source ~/.bashrc
查看环境变量是否配置成功:nvcc -V。成功会显示CUDA信息。