딥러닝 환경 한방설치스크립트

#!/bin/bash

set -e

### NVIDIA DRIVER(525.147.05) + CUDA(12.3) + CUDNN(8) + DOCKER 설치 스크립트
### Tested on Ubuntu 22.04 + A10 GPU

echo "[0/8] Updating apt..."
sudo apt-get update

echo "[1/8] Installing NVIDIA Driver..."
sudo dpkg -i ./nvidia-driver-local-repo-ubuntu2204-525.147.05_1.0-1_amd64.deb
sudo cp /var/nvidia-driver-local-repo-ubuntu2204-525.147.05/*.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get install -f -y
sudo apt-get install -y nvidia-driver-525

echo "[2/8] Installing CUDA Toolkit and cuDNN..."
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /"
sudo apt-get update
sudo apt-get install -y cuda-toolkit-12-3 libcudnn8 libcudnn8-dev

echo "[3/8] Setting up environment variables..."
echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc

echo "[4/8] Installing Docker..."
sudo apt-get install -y apt-transport-https ca-certificates curl gnupg-agent software-properties-common
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable"
sudo apt-get update
sudo apt-get install -y docker-ce docker-ce-cli containerd.io

echo "[5/8] Installing NVIDIA Container Toolkit..."
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update
sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker

echo "[6/8] Verifying nvidia-smi in container..."
docker run --rm --gpus all nvidia/cuda:12.3.0-base-ubuntu22.04 nvidia-smi || echo "⚠️ nvidia-smi 테스트 실패: 드라이버 적용 전일 수 있습니다."

echo " All done! 재부팅 후 NVIDIA 드라이버가 적용됩니다."
echo "재부팅: sudo reboot"