Windows WSL CUDA 설치

Windows WSL CUDA 설치

 

Windows WSL에 CUDA를 설치하기 위해 GPU 드라이버 설치를 진행한다.

아래 링크로 이동한다.

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

Get CUDA Driver 버튼을 클릭하여 Nvidia GPU driver를 다운로드 및 설치해준다.

 

GPU 카드를 선택한다.

 

설치 파일을 다운받아 실행하여 설치를 진행한다.

 

설치 이후 재시작을 진행한다.

 

WSL에 진입하여 해당 명령어로 CUDA 설치를 진행한다.

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin

sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda-repo-wsl-ubuntu-11-4-local_11.4.0-1_amd64.deb

sudo dpkg -i cuda-repo-wsl-ubuntu-11-4-local_11.4.0-1_amd64.deb

sudo apt-key add /var/cuda-repo-wsl-ubuntu-11-4-local/7fa2af80.pub

sudo apt-get update

sudo apt-get -y install cuda

nvidia-smi를 사용하기 위한 util 설치

$ sudo apt install nvidia-utils-515

 

GPU Driver 확인

$ nvidia-smi

Docker 설치 이후에 GPU 사용도 가능

$ docker run --gpus all nvcr.io/nvidia/k8s/cuda-sample:nbody nbody -gpu -benchmark

 

$ docker run --rm --gpus all nvidia/cuda:11.7.1-base-ubuntu18.04 nvidia-smi

감사합니다.