카테고리 없음

아자아자2편-GPU 설정

강블루 2024. 2. 19. 07:59

 

<GPU설정편>

 

Q1) 사내 서버에서 GPU를 사용하려고했던 경우


A1) 

https://pytorch.org/get-started/previous-versions/ 에 들어가서 어떤 버전을 설치해야 할지가 가장 고민이었는데

여러 블로그를 참고해 보니 보통 다 nividia-smi를 출력 후 나온 Version보다 낮은 걸 받길래 처음에는 낮은걸 받았다가 torch 버전이 너무 낮아 호환성 문제로 상위 버전을 돌려봤는데 돌아갔음 이건 추후에 공부 예정

 

<순서>

1) cuda 사용 여부 확인

ex)

import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)

2) 터미널에서 nividia-smi를 출력 후 Version 정보 확인

3)  https://pytorch.org/get-started/previous-versions/ 에 들어가서 그에 맞는 설치 정보를 들고옴

ex) conda install pytorch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 pytorch-cuda=11.8 -c pytorch -c nvidia

4)  cuda 사용 여부 확인

 

+꿀팁 
1) watch -n 1 nvidia-smi (실시간 GPU 상태 확인) 끌떄는 마찬가지로 ctrl+c
2) which python
3) kill -9 PID

 


 

 

나는 왜 이렇게 못할까? 이걸 왜 못하지 바보인가? 포기할까? 그런 생각 금지X

우리는 할 수 있다 아자아자!!