tensorflow2.4.0 버전에서 cublas 핸들 생성 실패: CUBLAS_STATUS_NOT_INITIALIZED 오류를 해결하는 방법

이 방법은 https://github.com/tensorflow/tensorflow/issues/45070 을 참조합니다.

기본 코드에 추가

physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True)

이 오류는 다음을 통해 해결될 수 있습니다.

아주 간단하다