[CUDA + cuDNN + PyTorch for Win10설치]

 

그래픽카드 : GTX 1060Ti 3GB 

CUDA : 10.1

cuDNN: Download cuDNN v8.0.5 (November 9th, 2020), for CUDA 10.1

PyTorch: Stable(1.7.1) + Windows10 + Conda + Python + CUDA 10.1

 

 

 

 

 

cuDNN 압축해제 후 해당 폴더 내에 있는 모든 파일을 복사 

 

  

복사한 모든 파일을 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1 에 붙여넣기

 

 

맨 아래 환경변수 선택

 

시스템 변수 > 'Path' 선택 후 > '편집' 선택

 

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvpp  

등록 여부 확인

 

 본인은 Window10 그리고 아나콘다, 파이썬 그리고  CUDA 는 10.1 버전을 사용하고 있으므로 다음과 같이 선택한다. 

아래의 Run this Command 란에 나타나 있는

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch

명령어를 아나콘다 프롬프트를 통해 실행시켜 준다.