여러 종류의 cuda setup

상황

cuda를 2가지 이상 install 해야하는 상황

방법

driver install은 2가지 중 하나를 실행하면 된다.

driver install 1

sudo ubuntu-drivers autoinstall

권장버젼으로 자동 설치

driver install 2

ubuntu-drivers devices
nvidia-smi

vendor   : NVIDIA Corporation
driver   : nvidia-driver-530 - third-party non-free
driver   : nvidia-driver-535-server-open - distro non-free
driver   : nvidia-driver-535-server - distro non-free
driver   : nvidia-driver-525 - third-party non-free
driver   : nvidia-driver-535-open - third-party non-free
driver   : nvidia-driver-525-open - distro non-free
driver   : nvidia-driver-535 - third-party non-free
driver   : nvidia-driver-545 - third-party non-free
driver   : nvidia-driver-525-server - distro non-free
driver   : nvidia-driver-545-open - third-party non-free recommended
driver   : xserver-xorg-video-nouveau - distro free builtin

필자의 경우는 다음과 같은 list가 있었음

출처: https://docs.nvidia.com/deploy/cuda-compatibility/
필요한 driver의 버젼은 위의 표를 참고하여 설치하면 된다.

sudo apt -y install nvidia-driver-525

필자의 경우 525를 사용하였다.
이후 nvidia-smi를 통해 설치가 잘 이루어 졌는지 확인

만약 작동이 되지 않을 경우
리부트(sudo reboot now)를 통해 해결이 될 수도 있음.

삭제 후 재설치

만약 원하지 않는 driver를 설치하여 재설치를 원할 경우 아래의 명령어를 통해 삭제할 수 있다.

sudo apt install build-eseential
sudo apt-get remove --purge nvidia-*
sudo apt-get autoremove
sudo apt-get update

드라이버 업데이트 tip

PPA는 런치패드에서 제공하는 우분투의 공식 패키지 저장소에 없는 서드 파티 소프트웨어를 위한 개인용 소프트웨어 패키지 저장소이다.

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

위의 ppa를 진행하고 apt update를 실행하면 nvidia-driver에 대한 update도 수행할 수 있게 된다.

cuda install

필자의 경우 cuda-11.8, cuda-12.3을 사용함.
본인의 서버에 맞는 정보를 입력한 후 Base Installer을 따라서 진행한다.

만약 2개의 cuda모두를 설치했다면
ls /usr/local/ | grep cuda를 입력했을 때,
아래와 같이 cuda가 모두 설치된 것을 확인할 수 있다.

cuda
cuda-11.8
cuda-12
cuda-12.3

nano ~/.bashrc입력

export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH

위 코드를 입력하고 저장한 후 나온다.
(이 때 cuda-11.8은 사용할 cuda 버젼에 맞추어 입력한다.)

source ~/.bashrc를 통해 bash를 다시 실행해준다.

이후 사용 용도에 따라 앞서 입력한 bashrc의 cuda를 바꾸어가면서 사용하면 여러개의 cuda를 사용할 수 있게 된다.

nvcc -V를 입력하여 cuda 버젼 확인 가능