Blas SGEMM launch failed issue
tf.import\_graph\_def
함수를 사용하였는데, InternalError:Blas SGEMM launch failed issue 가 발생하였다.
이 문제는 이미 작동중인 GPU 할당
에 관한 문제였는데,
주요 해결 방법으로
gpu_option
에서allow_grouth=True
를 설정하는 방법config.allow\_soft\_placement = True
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.0
과 tensorflow-gpu==1.12.0
version을 사용하고 있었는데,Cuda9.0
의 버그를 해결하는 Patch 1~4
를 모두 설치하는 것이었다.
다음의 페이지에 들어가면 Cuda Patch를 다운받을 수 있다.
다운받았던 Patch1
이 설치가 되지 않았는데, 이것에 대한 해결법은 여기에서 볼 수 있다.
'Error' 카테고리의 다른 글
CUDA 9.0 patch 1 설치 에러 (0) | 2020.11.24 |
---|---|
ModuleNotFoundError: No module named 'tensorflow_core.keras' 해결 (0) | 2020.07.16 |