cmd에서 nvidia-smi
cuda-version 확인 후 맞는 버전 설치
필자의 경우 Geforce GTX 1070 Ti
(그래픽카드 장치관리자에서 디스플레이어댑터 항목으로 가능)
cuda toolkit 11.7 downloads 구글 검색 후 설치
cuDNN 은
Download cuDNN v8.5.0 (August 8th, 2022), for CUDA 11.x
버전 다운로드
cuDNN 압축 해제 후 bin, lib, include 파일
CUDA toolkit 설치된 파일 경로에 붙여넣기
환경변수 편집 bin, lib, include 경로 path 새로 추가하고 재부팅
cmd에서 ncvv -V
후 cuda 출력보이면 성공
conda 가상환경 만든 후
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
설치
import torch
print(torch.cuda.is_available()) 후 결과 확인
참고한 사이트
https://anaconda.org/nvidia/cuda-toolkit
쿠다 설치
https://jeo96.tistory.com/entry/Pytorch-CUDA-cuDNN-%EC%84%A4%EC%B9%98-windows-10
-anaconda cuda 설치
https://yjs-program.tistory.com/199