728x90
위 사진과 같은 에러가 발생했다.
필자의 tensorflow 버전은 2.9.1이다. 그리고 CUDA Version은 11.4이다.
여기저기 오류에 대한 해결책을 찾아보니 CUDA 버전과 tensorflow-qpu버전이 맞지 않아서라고 떴다. 버전을 맞춰주기 위해서 아래 링크를 참고하여 Tensorflow를 맞춰주려하였으나, 내 CUDA version과 맞는 tensorflow-qpu 버전이 없었다.
https://www.tensorflow.org/install/source#tested_build_configurations
소스에서 빌드 | TensorFlow
Google I/O is a wrap! Catch up on TensorFlow sessions View sessions 소스에서 빌드 소스에서 TensorFlow pip 패키지를 빌드하고 Ubuntu Linux 및 macOS에 설치합니다. 명령어는 다른 시스템에도 적용될 수 있지만, Ubuntu 및
www.tensorflow.org
그래서 추가적인 해결책을 좀더 찾아보다가 발견한 또다른 에러 해결 방법은 1.XX버전으로 tensorflow를 다시 까는 것이었다. 1.XX버전은 깔리지가 않았다.
tensorflow v2에서 tensorflow v1으로 변경하는 코드를 추가하였다.
코드를 추가하고 다시 재실행하니 정상적으로 실행되었다.
728x90
