리눅스 GPU 서버 설정

# NVIDIA 드라이버 설치
sudo ubuntu-drivers devices
sudo apt install nvidia-driver-550 # for GeForce RTX 2080 Ti

# 특정 버전의 cuda 설치 (e.g. cuda12.1)
wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
sudo sh cuda_12.1.0_530.30.02_linux.run

# 여러 버전의 cuda 관리
# 버전 확인
sudo update-alternatives --query cuda
# 12.1 등록
sudo update-alternatives --install /usr/local/cuda cuda /usr/local/cuda-12.1 121
# 등록 확인
ls -l /usr/local/cuda
# 버전 전환
sudo update-alternatives --config cuda


# shell 설정 파일(e.g. .zshrc)에 환경변수 추가
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
export CUDA_HOME=/usr/local/cuda
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

# 버전 확인
source ~/.zsrhc
nvcc --version