tensorflow1.0版本模型运行中出现error:Blas GEMM launch failed
时间: 2024-02-18 15:44:45 浏览: 244
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社区,获得更好的解决方案。
相关问题
tensorflow1.0版本模型运行中出现error:Internal: Blas GEMM launch failed
这个问题通常是由于GPU内存不足或者GPU驱动版本不兼容导致的。可以尝试以下解决方案:
1. 减少批量大小,使用更小的模型或者更小的输入图片尺寸,以减少GPU内存占用。
2. 检查GPU驱动版本是否与TensorFlow版本兼容,如果不兼容则升级或降级GPU驱动版本。
3. 如果使用的是CUDA版本的TensorFlow,则可以尝试升级CUDA版本,或者降级TensorFlow版本。
4. 如果以上方法都无法解决问题,则可能需要更换更高内存的GPU设备。
tensorflow.python.framework.errors_impl.internalerror: blas gemm launch failed
这个错误是由于 TensorFlow 在执行矩阵乘法时调用了 BLAS 库,但是 BLAS 库出现了错误导致无法执行。可能的原因包括 BLAS 库未正确安装或配置、硬件问题或其他系统问题。建议检查 BLAS 库的安装和配置,并确保硬件和系统环境正常。
阅读全文