[AI] Tensorflow GPU 환경 설정

설치 참고 링크 : https://hyunhp.tistory.com/740


실습 환경

python : 3.8.20

tensorflow : 2.9.0 ( tensorflow-gpu 는 tensorflow 2.10 이하에서만 운용 가능)


CUDA 관련 설치

cuda : 11.2.2

cudnn : 8.1.1

- cudnn 설치 시, 내부에 있는 bin, include, lib 폴더의 내용물을 복사


cuda 설치한 경로(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2)에 내용물을 각각의 폴더에 덮어쓰기로 적용


환경변수 Path 에 대해 각각의 변수 추가

- bin

- include

- lib, x64(x64의 경우 변수를 추가해도 안되는 경우 적용)


tensorflow-gpu 설치

- pip install tensorflow-gpu==2.10.0


의존성 관련 주요 패키지 버전

protobuf : 3.20.3

keras : 2.9.0 

tensorflow-estimator : 2.9.0

tensorboard : 2.9.0

numpy : 1.22.4

flatbuffers : 1.12

absl-py : 0.15.0


※ tensorflow, protobuf 에 대해 의존성 충돌발생시,

pip install --force-reinstall tensorflow==2.9.0 protobuf==3.20.3


24/10/09 - Could not load dynamic library 'cudart64_110.dll'; dlerror:cudart64_100.dll not found
cudart64_110.dll 파일 탐색이 안되는 경우 발생
- CUDA Toolkit 재설치 또는 vc_redist_x64.exe 설치
→  vc_redist_x64.exe 설치로 문제 해결(재부팅 필수)

댓글