#!/bin/bash
# GPU Setup Script - NVIDIA Driver + CUDA Installation
# Task: task-551.1 (작업 A)
# 이 스크립트는 sudo 권한이 필요합니다. 터미널에서 직접 실행하세요.
# 실행: sudo bash /home/jay/workspace/scripts/setup-gpu.sh

set -e

echo "=== GPU Setup: NVIDIA Driver + CUDA Installation ==="
echo ""

# Step 1: Blacklist nouveau
echo "[1/4] nouveau 블랙리스트 설정..."
cat > /etc/modprobe.d/blacklist-nouveau.conf <<EOF
blacklist nouveau
options nouveau modeset=0
EOF
echo "  → /etc/modprobe.d/blacklist-nouveau.conf 생성 완료"

# Step 2: Update initramfs
echo "[2/4] initramfs 업데이트..."
update-initramfs -u
echo "  → initramfs 업데이트 완료"

# Step 3: Install NVIDIA driver
echo "[3/4] NVIDIA 드라이버 설치..."
apt update -qq
ubuntu-drivers install
echo "  → NVIDIA 드라이버 설치 완료"

# Step 4: Install CUDA Toolkit
echo "[4/4] CUDA Toolkit 설치..."
apt install -y nvidia-cuda-toolkit
echo "  → CUDA Toolkit 설치 완료"

echo ""
echo "=== 설치 완료 ==="
echo "재부팅 후 아래 명령으로 확인하세요:"
echo "  /home/jay/workspace/scripts/verify-gpu.sh"
echo ""
echo "재부팅 명령: sudo reboot"