NVIDIA GeForce RTX 3070 GPU 설정 버전 정리

Version 정리

Windows 10 Home
Python 3.8.11
CUDA 11.2
cuDNN 8.1.1
tensorflow-gpu 2.5.0

자세한 설치 방법은 다음 링크에 잘 나와있다 : GPU 사용을 위한 CUDA 환경 구성하기

1. NVIDIA driver 설치

https://www.nvidia.co.kr/Download/index.aspx?lang=kr

설치확인

nvidia-smi

여기에 CUDA 12.2 라고

2. CUDA 11.2 설치

https://developer.nvidia.com/cuda-11.2.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

3. cuDNN 8.1.1 다운

https://developer.nvidia.com/rdp/cudnn-archive

nvcc --version

4. 가상환경 생성

가상환경 생성

conda create -n 가상환경이름 python=3.8.11
pip install tensorflow-gpu==2.5.0

5. GPU 확인

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())