[WSL2 Ubuntu] CUDA, CUDNN 설치

오늘은 Window에서 WSL2를 활용해서 Ubuntu 환경에서 AI 개발을 위한 환경을 구축할때, Nvidia 라이브러리인 Nvidia-Driver, CUDA, CUDNN을 설치하는 방법을 정리해보겠습니다.  

 

-목차-

1. Nvidia-Driver 설치
2. CUDA 설치
3. CUDNN 설치
4. 시스템 환경 변수 추가
5. 설치 확인

 

---

 

- 사전 작업: nouveau 드라이버를 블랙리스트에 추가

# nouveau 드라이버를 블랙리스트에 추가
sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
sudo bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf"
 
# 아래 명령어로 blacklist-nvidia-nouveau.conf 파일 내용 확인
cat /etc/modprobe.d/blacklist-nvidia-nouveau.conf
blacklist nouveau
options nouveau modeset=0
 
# 아래 명령어 입력 후 재부팅
sudo update-initramfs -u
 
# nouveau 실행 확인 -> 아무것도 안뜨면 종료된 것
lsmod |grep nouveau

1. Nvidia-Driver 설치 : WSL 환경에서는 윈도우에 설치된 nvidia driver를 사용하므로 Ubuntu 내부에서 nvidia driver를 설치하면 안된다.

• 'users must not install any NVIDIA GPU Linux driver within WSL 2'이라고 강조되어 있다. [2]
• 근데 nvidia-smi를 쳐도 안뜨는 경우 nvidia-utils-버전 라이브러리를 설치해야함.

  sudo apt install nvidia-utils-550-server
  ​

2. CUDA 설치

• Cuda archive 홈페이지 접속 후 원하는 설정에 맞는 설치 파일 다운 [3]

• Cuda 설치 확인

  ls /usr/local/ | grep cuda
  
  cuda
  cuda-12.3
  ​

3. CUDNN 설치

• CUDNN Archive 홈페이지에서 원하는 버전 다운 [4]

• cudnn tar file 압축풀기

tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz

• cudnn include 복사 붙여넣기

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

• cudnn lib64 복사 붙여넣기

sudo cp lib/libcudnn* /usr/local/cuda-12.3/lib64

 

4. 시스템 환경변수 추가

bashrc 수정: sudo vi ~/.bashrc

업데이트: source ~/.bashrc

 

5. 설치 확인

• nvcc -V, nvidia-smi 출력 확인

 

Reference

[1] https://velog.io/@cjkangme/WSL2로-CUDA-환경-설정하기-CUDAcuDNN-설치까지

[2] https://docs.nvidia.com/cuda/wsl-user-guide/index.html

[3] https://developer.nvidia.com/cuda-toolkit-archive

[4] https://developer.nvidia.com/cudnn-archive