Tensorflow GPU, CUDA, cuDNN 아나콘다에서 설치하기

최근 Windows 11로 넘어갔다가 장렬히 망했습니다.

컴퓨터 포맷 후 다시 데이터 분석용 환경을 구축한 과정을 공유합니다.

 

시작하기 전 그래픽 카드 등 하드웨어 적합하며

TensorFlow GPU 2.5 버전을 사용한다는 전제로 진행했습니다.

 

목표

TensorFlow GPU 2.5 버전 설치
TF GPU 2.5 사용을 위해 필요한 CUDA 11.2, cuDNN 설치
TF GPU 2.5를 사용할 환경 생성
아나콘다 설치

 

 

바로 드가입시다.

 

아나콘다 설치

아나콘다 다운로드 홈페이지 바로가기 - Anaconda Distribution

Anaconda | Anaconda Distribution

 

아나콘다 설치는 어렵지 않으니 링크 첨부에서 마치겠습니다.

 

 

 

Anaconda Prompt 

 

 

환경 생성

conda create -n OOOO python=3.8

파이썬 3.8버전으로 갑니다.

 

OOOO 에 원하는 이름 넣으세요.

저는 DA로 했습니다.

 

 

Proceed 물으면 y

 

---

 

생성한 환경 활성화

conda activate OOOO

 

---

 

TensorFlow-GPU 2.5.0 설치

pip install tensorflow-gpu==2.5.0

---

CUDA 11.2 및 cuDNN 설치

conda install -c conda-forge cudatoolkit=11.2 cudnn

 

 

---

파이썬에서 설치 확인

python

import tensorflow as tf

print("GPU Available: ", tf.config.list_physical_devices('GPU'))
print("CUDA Version: ", tf.sysconfig.get_build_info()["cuda_version"])
print("cuDNN Version: ", tf.sysconfig.get_build_info()["cudnn_version"])