설치순서
1. 그래픽카드 최신 드라이버 깔기 (nvidia 홈페이지)
2. visual studio 2019 (community version)
3. CUDA 11.1 (_456.43 _ win10) 설치
4. cuDNN-v8.0.5 (windows x64) >> 압축해제 후 CUDA 설치 디렉토리에 맞춰 파일 복사
** CUDA 설치 디렉토리(기본) : Program Files / NVIDIA GPU Computing Toolkit / CUDA
A. tensorflow 테스트
<tensorflow-gpu 2.4.0 version>
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
다음 코드 실행해보니
대략...
1. GPU는 확인됨
2. 근데 라이브러리가 딴건 다 있는데 cusolver64_10.dll이 없어서 에러가 난다는 말.
3. 그래서 GPU 등록을 skip한다는거
그래서 문제의 그 라이브러리를 보기 위해, CUDA 설치 폴더의 bin 디렉토리를 보았다!
그리고 다시 실행.
만세! 잘된다.
B. PyTorch Test
(아나콘다 가상환경에서 시작)
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
패키지 쪼끔 더 설치
conda install jupyter notebook ipython
그리고 코드 테스트~
잘된다 잘된다~~