* Tensorflow 2.7 설치
- RTX3090은 CUDA 11버전 이상만 사용 가능, 따라서 Tensorflow 2.5 버전 이상만 사용 가능
- 현재 출시된 Tensorflow 버전은
https://www.tensorflow.org/install/source_windows#tested_build_configurations 에서 확인
- 2.7버전이 위 페이지에서는 확인 불가능하지만 실제 릴리즈되었으므로 2.7버전으로 설치
- Tensorflow 2.5버전은 Keras와 호환 문제가 있으며 Tensorflow 2.6버전은 업데이트 시기에 따라 Keras 2.6또는 2.7과 호환됨 (중간에 호환성 관련 오류를 수정하면서 버전이 서로 안 맞게 된 듯...)
- Tensorflow 2.7은 Keras 2.7과 호환되므로 Tensorflow 2.7을 설치
pip install tensorflow-gpu==2.7
- 위와 같이 설치하면 Keras 2.7도 함께 설치됨
- 케라스로 작성된 프로그램 실행 결과 아래와 같이 GPU를 사용하는 것으로 확인됨
댓글 없음:
댓글 쓰기