TensorFlow: InternalError: Blas SGEMM launch failed

TensorFlow: InternalError: Blas SGEMM launch failed

Asked on December 15, 2018 in Tensorflow.
Add Comment


  • 1 Answer(s)

    The solution for this is to do with where you place your weight and bias variables.

    Example:

    with tf.device("/job:paramserver/task:0/cpu:0"):
       W = weight_variable([input_units, num_hidden_units])
       b = bias_variable([num_hidden_units])
    
    Answered on December 15, 2018.
    Add Comment


  • Your Answer

    By posting your answer, you agree to the privacy policy and terms of service.