본문 바로가기

컴퓨터

[Tensorflow] Blas xGEMV launch failed 오류 해결법

텐서플로우를 쓰던 중 Blas xGEMV launch failed 라고 하면서 오류가 떴다. 구글에 검색해보니 이대로는 나오지 않았고 Blas GEMM launch failed로 검색어를 고쳐줬다. xGEMV나 GEMM이나 비슷해보이긴 했으나 일단 난 다르다고 생각하는 중이었다. 스택오버플로우에서  GEMM에 대한 해결책이 있길래 속는 셈 치고 해봤는데 됐다. 

import tensorflow as tf
physical_devices = tf.config.list_physical_devices('GPU') 
for device in physical_devices:
    tf.config.experimental.set_memory_growth(device, True)

출처: https://stackoverflow.com/questions/43990046/tensorflow-blas-gemm-launch-failed