[WSL] 우분투 22.04에 CUDA 설치하기

NVIDIA 드라이버 설치 후

 

CUDA Toolkit 설치하기(버전 12.2)

wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda

참조 : CUDA Toolkit 12.2 Update 2 Downloads | NVIDIA Developer

 

cuDNN 설치하기(버전 8.9.4.25)

sudo apt-get install libcudnn8=8.9.4.25-1+cuda12.2
sudo apt-get install libcudnn8-dev=8.9.4.25-1+cuda12.2
sudo apt-get install libcudnn8-samples=8.9.4.25-1+cuda12.2

 

※설치가이드 :  Installation Guide - NVIDIA Docs

Installation Guide - NVIDIA Docs

 

만약 설치가이드에 따라 cuDNN을 설치한 후 테스트 코드 컴파일 중 "FreeImage.h" 파일이 없다는 에러가 나오면 다음과 같이 해당 라이브러리를 설치해준다.

sudo apt install libfreeimage3 libfreeimage-dev

다음과 같이 'Test passed!'가 출력되면 성공

 

터미널에서 nvidia-smi 라는 CLI 유틸리티를 실행하면 GPU의 상태를 다음과 같이 표시해준다.

 

※일단 최신버전을 기준으로 설치하고 C로 컴파일이 성공하는 것 까지만 확인했는데, 실제로 Tensorflow를 사용해 개발 환경을 구축하는 것은 별개의 문제다.

사용하게 될 Tensorflow, TensorRT, 기타 라이브러리 버전에 따라 CUDA와 cuDNN의 버전을 매치시키는 작업이 필요하다.