linux 18.04 + geforce 2080 ti Pytorch gpu 사용 환경 구축 (with anaconda)

이전에 구축했을 때는 온갖 삽질을 다 해서 했는데, GPT 와 함께하니 좋은 설정들을 알려주면서 해서 정말 시간 절약이 많이 되었다.

그래도 미리 알고 있는거랑 아닌거는 차이가 많이 나니, 정리를 해보도록 하면,

 

1. nvidia driver 470.223 이미 installed.

 

2. Cuda version -> nvidia-smi 에서 뜨는것은 아마도 추천 버전인 듯 했다.

sudo apt install cuda-toolkit 을 하니.. (좀 다를수도 있다) 11.3이 뜨는데 그것도 괜찮을거같아서 install 했다.

pytorch 홈페이지에 나온 커맨드

3. nvcc --version으로 cuda version 확인

 

4. pytorch 버전 확인 -> gpt 에 의하면 나의 환경에 맞는 pytorch는 1.12.1 이라고 한다. 왠지는 잘 모르겠지만.. 

pytorch로 가서 conda install 하는 command를 복사해온다.

 

근데 하려다 보니깐 생각해보니 conda 에서 해야되는데, python은 몇으로 해야되는지 또 고민된다. gpt한테 물어보니 3.10으로 하는게 좋을것이라고 해서 3.10 으로 conda environment create해서 진행한다.

 

이 외에도 여러 문제들이 있었지만.. (numpy downgrade 이슈 등..)

 

결국 torch를 gpu 에서 사용하는데 성공합니다.

validate하는 코드:

import torch
print(torch.cuda.is_available())  # Should return True
print(torch.cuda.get_device_name(0))  # Should display 'NVIDIA GeForce RTX 2080 Ti'