TensorFlow 버전호환성체크

loci·2024년 12월 28일

AI

목록 보기
1/1

TensorFlow GPU사용을 위해 CUDA와 cuDNN을 설치했는데 인식되지 않는다.
알고보니 버전이 호환되지 않아서 생긴 문제였다.
현재 나는 TensorFlow 2.18.0 버전에 CUDA 12.7버전을 설치했었는데 12.5를 설치했어야 했다.

https://www.tensorflow.org/install/source?hl=ko#gpu

위 사이트를 통해 호환되는 버전을 미리 체크하고 설치해야한다.

CUDA 다운: https://developer.nvidia.com/cuda-toolkit-archive
cuDNN 다운: https://developer.nvidia.com/cudnn-archive

수정: TensorFlow 2.10 이후 버전부터는 윈도우 환경에서 GPU를 사용할 수 없다고 한다. https://namu.wiki/w/TensorFlow (3번 항목 참고)

profile
편리한 개발자

0개의 댓글