[VScode] code에서 cuda 사용하기

우선 본 포스팅은 code와 python은 깔려있다고 가정한다.

 

보통 아래와 같은 식으로 활성화 여부를 확인할 수 있다.

if not torch.cuda.is_available():
    print("CUDA is not available. Exiting the program.")
    exit()
device = torch.device("cuda")

 

만일, CUDA가 활성화 상태가 아니라면 아래와 같은 방법으로 해결할 수 있다.

 

#1. 그래픽카드, CUDA 버전 확인

명령어 nvidia-smi or nvcc --version를 통해 확인 가능하다.

 

 

#2.  cuda version과 호환되는 pytorch 설치

아래 링크에 들어가서 내 cuda 버전과 일치하는 pip 명령문을 가져오면 된다.

https://pytorch.org/get-started/previous-versions/

Previous PyTorch Versions

 

 

알맞게 설치를 해주면 GPU로 잘 실행이 될 것이다!