엔비디아 글카 전용임
암드는 몰?루
념글에 WD1.4Tagger 이거 텐서 gpu로 구동 못하는 사람이 있는거 같아서
WD1.4Tagger 기준으로 글 싸봄
0. 엔비디아 그래픽 드라이버 설치
이건 지익스 깔아도 알아서 해주고
웬만하면 다 했을거 같으니 설명 필요없을듯
1. 쿠다 툴킷 설치
CUDA Toolkit 11.8 Downloads | NVIDIA Developer
지금 기준(22년11월18일)으로 쿠다 툴킷 최신버전이 11.8인데 텐서랑도 호환되는거 같으니 받아서 설치
2. cudnn 설치
cuDNN Download | NVIDIA Developer
여기서 다운 받으면 되는데
가입 귀찮을수도 있으니
aHR0cHM6Ly9tZWdhLm56L2ZpbGUvc2ZjaUNRUUwjZzNhZGtNXzBlSjlubjZ6dEp5NERHMkMtamsydkdXeU95dWdSVmdJQm9tTQ==
8.6.0 버전임
나중에 더 최신 버전 원하거나 링크 터졌으면 알아서 위 링크에서 받아주기
압축풀면 bin, include, lib 폴더가 나오는데
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\(쿠다버전) 으로 들어가서
거기에 그대로 폴더 덮어쓰면 됨
11.8 버전은
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8
여기임
3. ZLIB 설치
http://www.winimage.com/zLibDll/zlib123dllx64.zip
위험한 파일이라고 떠도 그냥 무시하셈
쫄리면 그냥 CPU 쓰삼
출처는 Installation Guide :: NVIDIA Deep Learning cuDNN Documentation 여기
압축풀어서 dll_x64에 들어가면
이 파일들이 나오는데
이 파일들 전부 혹은 zlibwapi.dll을
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\(쿠다버전)\bin
에다가 넣어주면됨
여기까지 잘 따라했다면
이제 파이썬 스크립트에서 쿠다툴킷 경로 잘 지정되어있다면
텐서플로 gpu 잘 구동될거임
텐서플로 gpu가 쿠다 툴킷만 깐다고 해서 다 되는게 아니라서
이거외에도 더 필요할수도 있는데
파이썬 스크립트 로드할때 에러 보면 어떤 파일이 부족한지 나오니까 그거 추가 설치해주면 됨