Error

Blas SGEMM launch failed issue 해결

Codezoy 2020. 11. 24. 19:29

Blas SGEMM launch failed issue

tf.import\_graph\_def 함수를 사용하였는데, InternalError:Blas SGEMM launch failed issue 가 발생하였다.

이 문제는 이미 작동중인 GPU 할당에 관한 문제였는데,
주요 해결 방법으로

  1. gpu_option에서 allow_grouth=True를 설정하는 방법
  2. config.allow\_soft\_placement = True
  3. config.log\_device\_placement = True

를 설정하는 방법이 나왔지만 효과가 없었다.
<예시 코드>

gpu_options = tf.GPUOptions(allow_growth = True)
    config = tf.ConfigProto(gpu_options=gpu_options)
    # config.gpu_options.allow_growth = True
    config.allow_soft_placement = True
    config.log_device_placement = True
    # GPU Set

    # eval_sess = tf.Session(graph=eval_graph)
    with tf.Graph().as_default() as graph:
        with tf.Session(graph=graph, config=config) as eval_sess:

나의 로컬PC는 Cuda9.0tensorflow-gpu==1.12.0 version을 사용하고 있었는데,
Cuda9.0의 버그를 해결하는 Patch 1~4를 모두 설치하는 것이었다.

다음의 페이지에 들어가면 Cuda Patch를 다운받을 수 있다.

다운받았던 Patch1이 설치가 되지 않았는데, 이것에 대한 해결법은 여기에서 볼 수 있다.