failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
Deep Learning/TensorFlow 2021. 4. 14. 14:36윈도우 10에 텐서플로우 2.x버전에서 gpu로 실행 시
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 에러 날 때는
import tensorflow as tf
config = tf.compat.v1.ConfigProto(gpu_options = tf.compat.v1.GPUOptions(per_process_gpu_memory_fraction=0.8)
#device_count = {'GPU': 1}
)
config.gpu_options.allow_growth = True
session = tf.compat.v1.Session(config=config)
tf.compat.v1.keras.backend.set_session(session)
를 코드 상단에 추가해주자.
github.com/tensorflow/tensorflow/issues/35887#issuecomment-679329549
