wsl에서 Pytorch Cuda버전 설치하기(+ VSCode)

* OS: Windows 11 - wsl2

* 윈도우 10은 wsl에서 host gpu에 접근이 불가능

  * Wsl 설치 방법 : https://webdir.tistory.com/541

 

* Wsl Cuda Driver 설치

* https://developer.nvidia.com/cuda/wsl

 

* pytorch_gpu 설치

  * Anaconda 설치

    * https://www.anaconda.com/products/individual 에서 Linux 버전으로 받은 후 sh파일 실행

   

  * CUDA version 체크

      $nvcc --version

     

     

   * https://pytorch.kr/get-started/revious-versions/ 에서 cuda에 맞는 pytorch 설치

 

* 설치 커맨드를 입력하면 저렇게 세부 설치 내용을 체크하는 부분이 있는데, 여기서 pytorch cpu가 아닌걸 꼭 확인해야 시간낭비를 안할 수 있습니다!!

 

* VSCode Juupyter Notebook extension 설치

      * ipynb 파일에서 파이썬 커널을 conda로 설정