CUDA, CuDNN 설치

호두마리·2023년 2월 12일
0

tensorflow

목록 보기
2/2

Tensorflow 버전 확인

현재 PC(또는 PC내 가상환경)에 설치된 Tensorflow 버전을 확인한다

Tensorflow 호환 CUDA, CuDNN 버전 확인

Tensorflow 공식 홈페이지에서 설치된 Tensorflow 버전과 호환되는 CUDA, CuDNN 버전을 확인한다
https://www.tensorflow.org/install/source_windows#tested_build_configurations

CUDA 설치

NVIDIA 홈페이지에서 Tensorflow와 호환되는 CUDA 버전을 다운로드

CuDNN 설치

NVIDIA 홈페이지에서 Tensorflow 및 CUDA와 호환되는 CuDNN 버전을 다운로드하여 설치
cuDNN Archive

CuDNN 라이브러리 CUDA 적용

다운로드 받은 CuDNN 라이브러리 압축파일을 풀어 CUDA가 설치된 경로에 덮어쓰기
CUDA 설치 경로: C:\Program Files\NVIDA GPU Computing Toolkit\CUDA{version}

profile
자고싶당

0개의 댓글