tensorflow InternalError (see above for traceback): Blas SGEMM launch failed : m=1228800, n=15, k=30

环境:

16.04.1-Ubuntu

python 3.6.10

tensorflow-gpu 1.12.0

CUDA Version 9.0.176

Cudnn 7.6.4

问题:

运行代码出现:InternalError (see above for traceback): Blas SGEMM launch failed : m=1228800, n=15, k=30

错误。

原因:最初的cuda 9.0.176版本有bug, 需要下载补丁进行修复。

补丁地址:

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal

参考文献:

https://github.com/qqwweee/keras-yolo3/issues/362

你可能感兴趣的:(Tensorflow)