감자

감자

병렬 프로그래밍 개발자

[Environment Setup] WSL2 환경에서 CUDA 설치하기

Date:     Updated:

카테고리:

태그:

🎯 WSL2 환경에서 CUDA 설치하기

일반적인 ubuntu 환경에서 설치하는 것과 비슷할 것 같으나 실제 설치한 순서대로 정리해보고자 한다.

  • 당연히 NVIDIA 그래픽 카드가 있는 것으로 가정한다.
  • 또한, 이 포스트는 모든 명령어를 최신 버전으로 설치한다는 가정하에 작성되었다.

Image

1. 📥 NVIDIA 드라이버를 Windows에 설치

WSL2에서 사용하지만 Windows 10에 제대로 드라이버가 설치되어 있어야 오동작을 안한다.

NVIDIA 드라이버 다운로드 페이지

1.1. 드라이버 설치 확인

  • 아래 명령어를 실행하여 드라이버가 설치되어 있는지 확인해보자.
  • 현재 내 Windows 10에는 드라이버가 설치되어 있지 않다.
nvidia-smi

Image

1.2. 드라이버 설치

  • NVIDIA 드라이버 다운로드 페이지에 접속하여 자신의 버전에 맞는 드라이버를 설치하자.
    • 나는 노트북에 GeForce RTX 2070을 사용하고 있으므로 그 버전에 맞는 드라이버를 설치한다.
  • 설치가 끝나면 재부팅을 해주자.

Image

1.3. 드라이버 설치 확인

  • 재부팅 후 아래 명령어를 실행하여 드라이버가 설치되었는지 확인해보자.
  • nvidia-smi 명령어를 입력하면 아래와 같이 나오면 NVIDIA 드라이버가 잘 설치되었다는 뜻이다.
nvidia-smi

Image

2. 📥 Windows에 CUDA Toolkit 설치

이제 드라이버가 설치되었으니 CUDA Toolkit을 설치해보자.

2.1. CUDA Toolkit 설치 확인

  • 아래 명령어를 실행하여 CUDA Toolkit이 설치되어 있는지 확인해보자.
  • 현재 내 Windows 10에는 CUDA Toolkit이 설치되어 있지 않다.
nvcc --version

Image

2.2 Visual Studio 설치

  • CUDA Toolkit을 설치할 때 빠른 설치로 진행하면 Visual Studio를 요구하는 설치가 포함되어 있다.
  • 스스로 어떤 것들을 제외해야할지 모르겠다면 마음 편하게 Visual Studio를 설치하자.

Image

2.3. CUDA Toolkit 설치

  • CUDA Toolkit 설치 페이지에 접속하여 자신의 버전에 맞는 CUDA를 설치하자.
    • nvidia-smi 명령어를 입력하면 아래와 같이 나오는데 여기서 CUDA Version과 동일한 CUDA Toolkit을 설치하면 된다.
PS C:\Users\KGD> nvidia-smi
Sun Mar  9 14:36:33 2025
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 572.60                 Driver Version: 572.60         CUDA Version: 12.8     |
|-----------------------------------------+------------------------+----------------------+
| GPU  Name                  Driver-Model | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
|                                         |                        |               MIG M. |
|=========================================+========================+======================|
|   0  NVIDIA GeForce RTX 2070      WDDM  |   00000000:01:00.0 Off |                  N/A |
| N/A   41C    P0             32W /  115W |       0MiB /   8192MiB |      0%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+

+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI              PID   Type   Process name                        GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|  No running processes found                                                             |
+-----------------------------------------------------------------------------------------+
  • 아래와 같이 CUDA Toolkit 설치 중간에 Visual Studio에 종속된 설치들이 있다.

Image

2.4. CUDA Toolkit 설치 확인

  • 아래 명령어를 실행하여 CUDA Toolkit이 설치되었는지 확인해보자.
nvcc --version

Image

3. 📦 WSL2에 CUDA 설치

CUDA 설치 페이지에 접속하여 자신의 버전에 맞는 CUDA를 설치하자.

  • WSL2에 NVIDIA 드라이버는 설치할 필요가 없다. Windows에 설치한 드라이버로 알아서 설정된다.
  • 위에 언급한 CUDA 설치 페이지를 가면 아래처럼 내 환경에 맞는 설치 방법을 알려준다!

Image

4. 🔧 환경변수 설정

~/.bashrc를 수정해야 하는 경우

echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc

~/.zshrc를 수정해야 하는 경우

echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.zshrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.zshrc
source ~/.zshrc

5. ✅ 최종 설치 확인

nvcc --version

6. 📝 참고 사항

  • cuda-repo-wsl-ubuntu의 앞으로 쓰지 않을 가능성이 높으므로 삭제한다.
rm cuda-repo-wsl-ubuntu-12-8-local_12.8.1-1_amd64.deb

Environment Setup 카테고리 내 다른 글 보러가기

최근 글 10 개 :)