[WSL] cuda toolkit 설치

sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo apt-get autoclean
sudo rm -rf /usr/local/cuda*

1. CUDA 설치.

간편한 방법이 있지만 그래도 한번 수동으로 설치해 보는 것도 좋으니 잘 따라오길 바란다.

이제 gpu 사용을 위해서 CUDA 드라이버를 설치하자.
이미 CUDA 에서도 wsl을 위해서 설치 가이드를 제공하고 있다. (링크)
우리에게 필요한 부분은 3.CUDA Support for WSL2 이부분이다.
일단 윈도우에 최신 혹은 원하는 드라이버를 설치한다. 리눅스에 드라이버를 설치 하지 않는다.
그리고 다음 명령어를 입력한다.

sudo apt-key del 7fa2af80

입력후 다음 주소( https://developer.nvidia.com/cuda-toolkit-archive ) 자신이 원하는 버전의 툴킷을 클릭한다.

다음과 같은 페이지를 볼 수 있다.

상단부터 하단까지 클릭하고 나면 마지막에 명령어들을 볼수 있다.
하단 명령어들을 vscode 터미널에 입력하여 설치 하자.
다운로드부터 설치까지 시간이 꽤 걸린다.

(옵션) 필요시 cuDNN 도 다운로드한다. ( https://developer.nvidia.com/rdp/cudnn-download#a-collapse714-92 ) -계정 필요. 

보이는 것처럼 우리가 선택한 CUDA 11.7 에 맞는 8.8.1 선택후 Linux x86_64 를 선택하여 다운 받는다. 용량이 상당히 크다.(900mb 정도)
윈도우에서 받은 파일을 Ubuntu로 넘겨야 되는데 vscode로 드래그 해도 되지만 윈도우 쪽에서 넣어주면 더 빨리 넣을수 있다.

상단의 그림에 보면 Ubuntu 를 접근 할 수 있고 home 폴더 밑에 계정명에 파일을 드래그 해서 넣어주면 된다. 이후
다시 vscode로 돌아와서 명령어 
tar xvf cudnn 입력후 tab을 눌러서 이름 자동완성 기능을 사용하여 입력한다.

그러면 좌측에 새로운 폴더가 생긴걸 확인할수 있다. 
파일들을 /usr/local/cuda-버전 위치에 복사 시키기 위해 해당 경로에 폴더가 잇는지 확인한다.

그림에 보듯이 cd 명령어를 사용해서 폴더를 옮겨 다닐수 있고/usr/local/ 하위 폴더로 cuda-11.7이 생성되있음을 알수 있다.
이후 cd /home/계정명 으로 다시 돌아와 다음 명령어로 파일들을 복사한다.

sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda-11.7/include 

sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda-11.7/lib64 

sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda-11.7/lib64/libcudnn*
-- 여기까지 (옵션)

2. 환경변수&경로 등록
cuda 설치가 완료되면 이제 파일의 경로를 리눅스에 등록해야한다.
좌측에 파일들 중에서 .bashrc 를 클릭한뒤 우리가 복사 했던 경로(/usr/local/cuda-11.7)를 추가한다.

수정후 ctrl + s 로 저장한뒤 명령어

 

source ~/.bashrc 

 

를 입력해서 리눅스에 등록을 완료한다. 
완료후 명령어
 
nvcc --version

입력해본뒤 아래와 같이 나오나 확인해보자.

nvcc: NVIDIA (R) Cuda compiler driver

Copyright (c) 2005-2022 NVIDIA Corporation

Built on Tue_May__3_18:49:52_PDT_2022

Cuda compilation tools, release 11.7, V11.7.64

Build cuda_11.7.r11.7/compiler.31294372_0

나온다면 성공 한것이다.
이제 리눅스에서 CUDA 사용이 가능해졌다.

 

3. 삭제 명령어

$ sudo apt-get purge nvidia*
$ sudo apt-get autoremove
$ sudo apt-get autoclean
$ sudo rm -rf /usr/local/cuda*