tensorflow1.0版本模型运行中出现error:Blas GEMM launch failed
时间: 2024-02-18 17:44:45 浏览: 261
tensorflow的模型
Blas GEMM launch failed 错误通常是由于GPU显存不足或者CUDA版本不兼容引起的。您可以尝试以下解决方法:
1. 减小batch size或者图像尺寸,降低显存占用。
2. 检查CUDA版本是否与TensorFlow版本兼容,如果不兼容,请升级或降低CUDA版本。
3. 如果您使用的是NVIDIA GPU,可以尝试升级NVIDIA驱动程序。
4. 尝试在TF中添加以下代码解决:
import tensorflow as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)
tf.keras.backend.set_session(session)
# 将上述代码添加到TF程序的开头即可
如果以上方法都无法解决问题,建议您咨询TensorFlow官方论坛或者GitHub社区,获得更好的解决方案。
阅读全文