PyTorch & CUDA Install

Pytorch 설치 과정에서 CUDA를 사용하게 되면 CUDA버전을 설정해야 한다.

맨 아래 CUDA Version

 

이미 내 PC에서는 GPU Driver가 설치되어 있었다.

GPU Driver 버전 확인은 cmd나 Anaconda Prompt에서 아래 명령을 실행하면 된다.

$ nvidia-smi

실행화면

 

그래픽 드라이버 설치는 따로 포스팅하지 않는다.

https://en.wikipedia.org/wiki/CUDA

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#cuda-major-component-versions

이런 사이트에서 잘 찾아서 해보길!!

 

위에 버전에 맞게 나는 이런식으로 조합하여 설치 명령을 추출해냈다.

참고로 Conda 환경에서 실행할 경우 패키지 매니저를 Conda로 바꾸면 된다. 나는 Visual Studio Code 환경에서 테스트 하기 위해 Pip으로 했다.

아래 사이트에서 다운로드 사이트 진입

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

최신 버전의 PyTorch를 사용하기 위해서는 Python 3.8 이상이 필요

※만약 gym, gymnasium 등을 사용할 경우 Python Version 3.8.x ~ 3.11.x 로 설치해야 한다.

아래 코드를 통해서 파이토치와 cuda가 정상적으로 설치 됐는지 확인한다.

import torch
x = torch.rand(5, 3)
print(x)
#tensor([[0.3380, 0.3845, 0.3217],
#        [0.8337, 0.9050, 0.2650],
#        [0.2979, 0.7141, 0.9069],
#        [0.1449, 0.1132, 0.1375],
#        [0.4675, 0.3947, 0.1426]])

import torch
torch.cuda.is_available()
#True