WSL2에서 NVIDIA-SMI 또는 CUDA 설정

WSL2에서 NVIDIA-SMI   설정하는 방법을 안내해드리겠습니다.

필수 조건 확인

먼저 다음 사항들을 확인해주세요:

• Windows 11 또는 Windows 10 (버전 21H2 이상)
• WSL2가 설치되어 있어야 함
• NVIDIA GPU가 설치된 시스템

설치 단계

1. Windows에서 NVIDIA 드라이버 설치

Windows 호스트에 최신 NVIDIA 드라이버를 설치해야 합니다:

• NVIDIA 공식 웹사이트에서 GeForce Game Ready 드라이버 또는 Studio 드라이버 다운로드
• 드라이버 버전이 472.12 이상인지 확인

2. WSL2에서 CUDA Toolkit 설치

WSL2 Ubuntu 환경에서 다음 명령어를 실행하세요:

 

bash

# 패키지 업데이트
sudo apt update

# 필요한 패키지 설치
sudo apt install -y wget gnupg

# NVIDIA CUDA 키 추가
wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-keyring_1.0-1_all.deb
sudo dpkg -i cuda-keyring_1.0-1_all.deb

# CUDA 저장소 추가
sudo apt update

# CUDA Toolkit 설치
sudo apt install -y cuda-toolkit

3. 환경 변수 설정

~/.bashrc 파일을 편집하여 PATH를 추가하세요:

 

bash

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

4. 설치 확인

다음 명령어로 설치가 올바르게 되었는지 확인하세요:

 

bash

nvidia-smi
nvcc --version

문제 해결

만약 nvidia-smi 명령어가 작동하지 않는다면:

1. WSL2 재시작: wsl --shutdown 후 WSL2를 다시 시작
2. Windows 드라이버 확인: Windows에서 NVIDIA 드라이버가 제대로 설치되었는지 확인
3. WSL 버전 확인: wsl -l -v로 WSL2가 사용되고 있는지 확인

주의사항

• WSL2에서는 별도의 NVIDIA 드라이버를 설치할 필요가 없습니다. Windows 호스트의 드라이버를 공유해서 사용합니다.
• Docker를 사용한다면 nvidia-docker2 대신 nvidia-container-toolkit을 사용해야 합니다.