CUDA 설치

·2022년 12월 29일
0
  1. cmd에서 nvidia-smi cuda-version 확인 후 맞는 버전 설치
    필자의 경우 Geforce GTX 1070 Ti
    (그래픽카드 장치관리자에서 디스플레이어댑터 항목으로 가능)

  2. cuda toolkit 11.7 downloads 구글 검색 후 설치

  3. cuDNN 은
    Download cuDNN v8.5.0 (August 8th, 2022), for CUDA 11.x
    버전 다운로드

  4. cuDNN 압축 해제 후 bin, lib, include 파일
    CUDA toolkit 설치된 파일 경로에 붙여넣기

  5. 환경변수 편집 bin, lib, include 경로 path 새로 추가하고 재부팅

  6. cmd에서 ncvv -V 후 cuda 출력보이면 성공

pytorch 설치

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

0개의 댓글