TensorFlow 2.3 GPU-InvalidArgumentError : 어설 션 실패 : [0] [Op : Assert] 이름 : EagerVariableNameReuse

Sep 25 2020

환경:

Windows 8.1 64 비트 Python 3.7.7 64 비트 TensorFlow 2.3 Keras 2.4

CUDA : 10.1 CuDNN : 7

오류 : InvalidArgumentError: assertion failed: [0] [Op:Assert] name: EagerVariableNameReuse.

문제 : model = Sequential()GPU 실행시 간단한 문 에서 위의 오류

그러나 GPU없이 스크립트를 실행하면 (CUDA / CuDnn 환경 변수를 설정하지 않고 예상대로 작동합니다.

그래서 효과적으로 GPU가 있지만 TensorFlow (2.3)와 함께 사용할 수 없습니다.

누군가 도울 수 있습니까?

답변

2 thushv89 Nov 11 2020 at 09:30

내가 본 것에서이 문제는 다음과 같은 치명적인 조합에 대해 나타납니다.

  • Windows OS
  • TensorFlow 2.3.x

나는 이것을 꽤 오랫동안 해결하기 위해 고군분투 해 왔으며 최근에야 알아 냈습니다.

가장 쉬운 수정

가장 쉬운 솔루션은 TensorFlow 2.4.0rcx를 사용하는 것입니다. 아직 시험판입니다. 그러나 나에게 앞서 언급 한 문제를주지 않았습니다. 그래서,

pip uninstall tensorflow
pip install tensorflow==2.4.0rc0

TF 2.3.x를 유지하려는 경우 솔루션

TensorFlow 2.3.x 사용에 대해 정말로 단호하다면 유일한 솔루션은 다음과 같습니다.

  • Linux 기반 OS (Ubuntu) 사용해보기
  • 소스에서 TensorFlow 컴파일 ( 이 유용한 주석에 따라)