CUDA & PyTorch 설치(23.08) (1)

Linux 환경안에 딥러닝 환경 세팅을 해보았다. 매번 업데이트하면 달라지기 때문에 이전버전(23.08)과 가장 최근에 설치했던 버전(24.01)을 올려본다.

딥러닝을 돌리기 위해서는 cuda설치가 필수적이다.

딥러닝의 대표적인 프레임워크는 텐서플로우, 파이토치이다. 그 중 파이토치 설치 하는 것을 기록하려한다.

🙋🏻‍♀️ 다운로드 받아야 할 목록

1. cuda
2. cuDNN
3. NVIDIA Driver

내 노트북 사양

1. NVIDIA GeForce RTX 3080 Laptop GPU

 

2. 지포스 RTX 3080 → 8.6

cuda 버전 최소 11.1 권장 ⇒ 11.2 설치함.

 

3. CUDA 11.2에 맞는 cuDNN 8.1.0 설치

해당 OS에 맞게 설치 ⇒ 나는 Windows

 

4. NVIDIA Driver 설치

컴퓨터 사양에 맞는 드라이버 설치 ⇒ NVIDIA Studio 드라이버 설치함.

 

5. 위에서부터 NVIDIA 드라이버, cuDNN, cuda

 

6. 제어판에서 NVIDIA관련된 모든것 삭제

 

7. NVIDIA 드라이버 설치(계속 다음 누르기)

 

8. cuda 11.2.0 설치

계속 다음 누르다가 설치가 진행 안될 경우에는 제어판 들어가서 NVIDIA Frameview SDK 삭제

 

9. cudnn 압축 풀고 압축 폴더에 있던 bin, include, lib 파일을

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2 여기에 붙여넣기

붙여넣기하면 시스템 환경변수에 자동으로 등록됨.

 

10. cmd창에 “C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2\extras\demo_suite\deviceQuery.exe”실행해서 PASS뜨면 완료.

 

 

가상환경 만들 때

v1.8.2 with LTS support

# CUDA 11.1 (Windows)

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c conda-forge

GPU는 깔려있으니 pythorch버전만 알면 된다.

LTS버전으로 깔기!