wsl2에 pytorch를 설치하려고 했으나 cuda 지원 버전이 잘 설치되지 않아서 찾아봤습니다.
나는 마침내 그것을 시도하고 해결했습니다.
물론 pytorch에서 cuda를 사용하기 위해서는 cuda를 지원하는 그래픽 카드가 필요합니다.
공식 사이트에서
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
엔터라고 하는데… pytorch가 GPU 버전이 아닌 CPU 버전으로 설치되었습니다.
아래에 많은 pytorch-cuda가 있어도 도움이되지 않는 것 같습니다 …
우선 리눅스에서는 별도의 그래픽 카드 드라이버를 설치할 필요가 없고 윈도우에서는 최신 버전의 그래픽 카드 드라이버만 설치하면 된다.
가상화 같은건 잘 켜고 wsl2가 깔려있으면 아마 문제없이 잡을겁니다.
nvidia-smi가 정상적으로 작동하면 잘 준비된 것입니다.
다음 명령을 사용하여 GPU 버전의 Pytorch를 설치합니다.
$ conda install -c conda-forge pytorch-gpu
리포지토리는 conda-forge입니다.
처음 pytorch 저장소로 선택했을 때 작동하지 않고 멈췄습니다.
설치해야 할 것은 pytorch-gpu입니다.
pytorch-cuda가 아닙니다.
파이썬을 실행하여
가져오기 토치 실행
torque.cuda.is_available()을 확인하고 True로 표시되면 설치가 성공한 것입니다.
이것은 내 컴퓨터에 설치된 버전입니다.