python 3.9 버전 cuda 설치하여 pytorch 연동하기

python 3.9 버전과 cuda 프로그램 설치하여 pytorch 연동시키기

이놈의 cuda 연동하려고 하니.. 환경변수랑.. 버전이랑.. 먼가 잘 작동이 안 된다.

yolo 구동시켜볼라다가 포기한 적이 있었다.

이번에 사진 해상도 올리는 코드를 발견했는데.. CPU로 돌리니깐.. 너무 오래 걸려서....

​

이번 기회에 한번 cuda 연동을 시켜 본다.

​

검색에 검색을 거듭하여... 파이썬 버전 cuda 버전... pytorch 버전 다 맞추어야 먼가 돌아가는 걸 깨닫고.. 천천히 해본다

​

대충 python 3.9 버전에 맞추어서 검색해 보았다..

​

https://blog.naver.com/choisboom/222591797579

이쪽 블로그에 내용을 참고해서 진행했다.

​

집의 그래픽카드는

지포스 3080 / 3080 / 3070

총 3개

​

​

1. 파이썬 3.9.9 버전을 설치

https://www.python.org/downloads/release/python-399/

 

다운로드할 버전

Python Release Python 3.9.9

The official home of the Python Programming Language

www.python.org

Windows installer (64-bit)

 

2. Visual Studio Community 2019 (version 16.11)

https://my.visualstudio.com/Downloads?q=visual%20studio%202019&wt.mc_id=o~msft~vscom~older-downloads

비주얼 스튜디오 커뮤니티 2019 버전을 설치!!!

​

3.CUDA Toolkit 11.2 Downloads​

https://developer.nvidia.com/cuda-11.2.0-download-archive?target_os=Windows&target_arch=x86_64

 

CUDA Toolkit 11.2 Downloads

Select Target Platform Click on the green buttons that describe your target platform. Only supported platforms will be shown. By downloading and using the software, you agree to fully comply with the terms and conditions of the CUDA EULA. Operating System Architecture Compilation Distribution Versio...

developer.nvidia.com

cuda_11.2.0_460.89_win10.exe

쿠다 툴킷 11.2 버전 다운로드

 

​

4. cuDNN Archive 다운로드

https://developer.nvidia.com/rdp/cudnn-archive

cuDNN Archive

Download releases from the GPU-accelerated primitive library for deep neural networks.

developer.nvidia.com

Download cuDNN v8.1.0 (January 26th, 2021), for CUDA 11.0,11.1 and 11.2

cuDNN Library for Windows (x86)

cudnn-11.2-windows-x64-v8.1.0.77.zip

11.2 버전

​

5. pytorch에서 설치 방법 알기

https://pytorch.org/get-started/locally/

PyTorch

An open source machine learning framework that accelerates the path from research prototyping to production deployment.

pytorch.org

 

Get Started 탭을 눌러서 쓰고 싶은 PyToch Build를 정하고 나의 OS windows를 선택하고 패키지는 pip를 사용 Python 3.9니깐 파이썬 선택 그리고 쿠다 버전도 선택해 줌

 

CUDA 11.3 버전 깔아도 되나 보다...

​

Stable에

windows

pip

python

CUDA11.3

선택하면 cmd창에 설치하는 명령어(Run this Command)를 가르쳐 준다

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

그대로 cmd 창 관리자 권한으로 열어서 설치해줌

 

그리고 파이썬을 실행시켜서 코드 쳐 봄

python

import torch torch.cuda.is_available

()

 

True로 나오면 드디어 GPU로 연산이 가능하다!!!

​

pip list 쳐 보면 pytorch 가 버전이 +cuda라고 적혀있는 게 보인다..

 

와.. 드디어.. 나도 CUDA를 사용할 수 있게 되었구나!!!! 크으으으으 감격

​

포맷하면 다시 세팅을 하기 위해 남겨 둔다.

python 3.9 버전 cuda 설치하여 pytorch 연동하기