WSL2 우분투 (24.04 LTS)에 한방에 GPU 연동하기

본인의 시스템 사양:

RTX A500, Intel Ultra7

sudo apt update
sudo apt upgrade

위 작업을 통해서 자고 있던 WSL를 일깨워주자.

sudo apt install nvidia-cuda-toolkit
nvcc --version

이제는 수동으로 다운로드할 필요가 없다. 이거 하면 바로 된다. nvcc --version을 통해서 설치되었는지 확인해보자. 안되었으면 수동하면 된다.

친절하게도 CuDNN도 자동으로 설치할 수 있다.

sudo apt install nvidia-cudnn

이제 아나콘다를 설치하고 알아서 환경을 구축하라. 본인의 경우 연구 환경 세팅은 파이썬 3.11이고, 프로덕트 레벨은 3.9다.

가상환경을 실행하고, 아래 명령문을 실행하자.

텐서플로우는

pip install tensorflow==2.15.0
python3 -m pip install --upgrade pip
python3 -m pip install wheel
python3 -m pip install --upgrade tensorrt

이렇게 설치하면 간편하게 모든게 설치된다. keras까지.