wsl 2 pytorch cuda gpu 설치

 

wsl2에서 pytorch 를 설치하려다가 cuda 지원되는 버전이 잘 설치가 안돼서 좀 찾아봤습니다.

 결국이리저리 시도 하다가 해결을 했습니다. 

pytorch에서 cuda를 사용하기 위해선 당연히 cuda가 지원되는 그개픽카드가 있어야 합니다. 

공식 홈페이지에선 

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

를 입력하라고 나옵니다만...pytorch 가 gpu 버전 대신 cpu 버전으로 설치돼버리더군요. 

저 중에 pytorch-cuda 라는게 별로 있어도 도움이 안 되는 듯...

 

일단, linux에 별개의 그래픽카드 드라이버를 설치할 필요는 없고, windows에서만 최신 버전의 그래픽 카드 드라이버를 설치하면 됩니다. 

가상화 같은것을 잘 켜고 wsl2가 설치되면 아마도 별 문제 없이 잡힐겁니다. 

 

nvidia-smi 가 정상적으로 작동한다면 준비는 잘 되었습니다. 

다음 명령어로 gpu 버전의 pytorch를 설치합니다.

$ conda install -c conda-forge pytorch-gpu

 레포지토리는 conda-forge 입니다. 처음에 pytorch 레포지토리로 선택했더니 제대로 안되고 얼어버렸습니다.

설치해야 하는 것은 pytorch-gpu입니다. pytorch-cuda 가 아닙니다. 

python을 실행해서 
import torch 를 실행하고

torch.cuda.is_available()을 확인해서 True가 나오면 정상적으로 설치된 것입니다. 

 

제 컴퓨터에 설치된 버전입니다.