텐서플로우를 쓰던 중 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
'컴퓨터' 카테고리의 다른 글
[Linux] 노트북 화면 닫아도 안 꺼지게 하기 (0) | 2022.01.22 |
---|---|
[Python] 주피터 노트북 커널이 자꾸 죽을때 (0) | 2022.01.22 |
[Linux] initramfs가 떴을때... 대처법 (0) | 2022.01.20 |
[Linux] 갑자기 리눅스가 멈췄을때 대응법(Magic SysRq Key) (0) | 2022.01.19 |
리눅스 &&, ||, &, ; 차이점 정리 (0) | 2022.01.04 |