WSL2에 ML 환경구축하기 (cuda, anaconda, pytorch) on window

1. CUDA 설치

  ※ 참고 : CUDA on WSL User Guide

 

 - window에 CUDA설치를 하고 WSL을 설치하면 자동으로 연동된다.

 - CUDA는 WSL2에서만 지원 가능하다.

 - WSL 버전 확인하는 커맨드 (cmd창에서 wsl -l -v)

 

2. WSL 설치방법 (on window)

  ※ 참고 : Install Linux on Windows with WSL

 

  - window powershell에 커맨드 입력

#처음으로 WSL을 설치하는 경우
wsl --install

#재설치하는 경우
wsl --install -d Ubuntu-20.04


#WSL 버전 확인하는 커맨드 
wsl -l -v

 

3. Anaconda 설치

 

1) 해당 링크에서 원하는 anaconda버전의 링크 주소를 복사한다.

       https://repo.anaconda.com/archive/

 

2) WSL 터미널에서 해당 커맨드를 입력한다.

# 다운로드 받기 - 복사한 링크 주소 입력
wget  https://repo.anaconda.com/archive/Anaconda3-5.3.1-Linux-x86_64.sh

# 해당 파일을 실행할 수 있게 허용해주기 위한 커맨드 - 복사한 링크 주소에서 가장 마지막 parse (파일명)
chmod +x  Anaconda3-5.3.1-Linux-x86_64.sh

# 파일 실행
sh Anaconda3-5.3.1-Linux-x86_64.sh

# 완료 후, 파일 제거
rm Anaconda3-5.3.1-Linux-x86_64.sh

 

3) 설치가 잘 되었는지 확인

 

   - window 재부팅 필요함

# 설치된 정보
conda info

# conda 내의 가상환경 정보
conda info--envs

 

4. Pytorch 설치

 

   - CUDA는 본인의 컴퓨터 사양에 따라 버전이 다를 수 있습니다. (버전이 다를 경우 에러 발생함)

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

 

   - python을 실행하여 pytorch가 잘 임포트되는지 확인

     (** 필자의 경우 OSError 발생함 --> libgompl 설치로 해결)

 

 

알고보니,,,,

cuda 11.7버전만 WSL2에서 지원되는 듯 하다...

torch.cuda.is_available() 코드가 자꾸 False...ㅠ

결국 듀얼부팅이 답인가...