Prerequisite
- 가상환경 ( 본 문서는 conda 로 진행됩니다 )
- CUDA 가 설치되어있다고 가정합니다.
가상환경 별 GPU 잡기?
A 가상환경에서는 torch gpu가 잡히는데 B 가상환경에서는 잡히지 않을리 없다.
이는 가상환경 별 torch 설치를 제대로 하지 못했을 가능성이 크다.
nvcc와 nvidia-smi 버전을 맞춘 후 (혹은 업그레이드) 아래 명령어를 입력한다.
다시 설치해주는 명령어
conda install pytorch=2.0 pytorch-cuda=11.8 -c pytorch -c nvidia
conda 설치 시 설치할 패키지, 빌드할 버전과 총 용량을 보여주는데 pytorch 부분에 cuda11.8이 명시되어있는지, conda에 깔린 python 버전과 같은 건지 꼭 확인한다
그리고 확인 !
python3 -c "import torch; print(torch.cuda.is_available())"