vkgs-sim / scripts /host_install_docker.sh

Add files using upload-large-folder tool

4c75661 verified 8 days ago

4.29 kB

#!/usr/bin/env bash
# 호스트에 Docker Engine + NVIDIA Container Toolkit 설치.
# sudo 비밀번호를 물어보므로 반드시 사용자가 직접 실행해야 합니다.
#   Claude Code 프롬프트에서:  ! bash scripts/host_install_docker.sh
set -euo pipefail

say()  { printf '\n\033[1;36m==> %s\033[0m\n' "$*"; }
ok()   { printf '\033[1;32m  ✔ %s\033[0m\n' "$*"; }
warn() { printf '\033[1;33m  ! %s\033[0m\n' "$*"; }
die()  { printf '\033[1;31m  ✘ %s\033[0m\n' "$*" >&2; exit 1; }

[[ $EUID -eq 0 ]] && die "root로 직접 실행하지 마세요. 일반 사용자로 실행하면 sudo를 알아서 씁니다."

. /etc/os-release
say "호스트: $PRETTY_NAME ($VERSION_CODENAME / $(dpkg --print-architecture))"
[[ "$ID" == "ubuntu" ]] || warn "Ubuntu가 아닙니다 — 이 스크립트는 Ubuntu 기준입니다."

command -v nvidia-smi >/dev/null || die "nvidia-smi 가 없습니다. NVIDIA 드라이버부터 설치하세요."
ok "$(nvidia-smi --query-gpu=name,driver_version --format=csv,noheader | head -1)"

say "sudo 권한 확인 (비밀번호를 물어볼 수 있습니다)"
sudo -v || die "sudo 실패"
# 설치가 길어지는 동안 sudo 타임아웃 방지
while true; do sudo -n true; sleep 50; kill -0 "$$" 2>/dev/null || exit; done 2>/dev/null &
SUDO_KEEPALIVE=$!
trap 'kill $SUDO_KEEPALIVE 2>/dev/null || true' EXIT

# ---------------------------------------------------------------- 1. Docker
if command -v docker >/dev/null && docker --version >/dev/null 2>&1; then
  ok "Docker 이미 설치됨: $(docker --version)"
else
  say "1/3  Docker Engine 설치"
  for p in docker.io docker-doc docker-compose podman-docker containerd runc; do
    sudo apt-get remove -y "$p" >/dev/null 2>&1 || true
  done
  sudo apt-get update -qq
  sudo apt-get install -y -qq ca-certificates curl gnupg
  sudo install -m 0755 -d /etc/apt/keyrings
  curl -fsSL https://download.docker.com/linux/ubuntu/gpg \
    | sudo gpg --dearmor --yes -o /etc/apt/keyrings/docker.gpg
  sudo chmod a+r /etc/apt/keyrings/docker.gpg
  echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $VERSION_CODENAME stable" \
    | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
  sudo apt-get update -qq
  sudo apt-get install -y docker-ce docker-ce-cli containerd.io \
                          docker-buildx-plugin docker-compose-plugin
  ok "설치됨: $(docker --version)"
fi

say "Docker 동작 확인 (hello-world)"
sudo docker run --rm hello-world >/dev/null && ok "Docker 런타임 정상"

# ------------------------------------------------- 2. NVIDIA Container Toolkit
if command -v nvidia-ctk >/dev/null; then
  ok "nvidia-container-toolkit 이미 설치됨: $(nvidia-ctk --version | head -1)"
else
  say "2/3  NVIDIA Container Toolkit 설치"
  curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey \
    | sudo gpg --dearmor --yes -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
  curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
    | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' \
    | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list > /dev/null
  sudo apt-get update -qq
  sudo apt-get install -y nvidia-container-toolkit
fi

say "docker 데몬에 nvidia 런타임 등록"
sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker
sleep 3

say "GPU 패스스루 확인"
sudo docker run --rm --gpus all nvidia/cuda:12.1.1-base-ubuntu22.04 nvidia-smi \
  | sed -n '1,12p' && ok "컨테이너에서 GPU 보임"

# --------------------------------------------------------- 3. docker 그룹
say "3/3  $USER 를 docker 그룹에 추가"
if id -nG "$USER" | tr ' ' '\n' | grep -qx docker; then
  ok "이미 docker 그룹 소속"
else
  sudo usermod -aG docker "$USER"
  warn "그룹 변경은 '재로그인' 후에 적용됩니다."
  warn "그때까지 이 프로젝트 스크립트들은 자동으로 'sudo docker' 로 폴백합니다."
fi

say "완료"
cat <<'MSG'
  다음 단계:
    bash scripts/build.sh     # 개발 이미지 빌드
    bash scripts/run.sh       # 컨테이너 기동
    bash scripts/shell.sh     # 셸 진입
MSG

	#!/usr/bin/env bash
	# 호스트에 Docker Engine + NVIDIA Container Toolkit 설치.
	# sudo 비밀번호를 물어보므로 반드시 사용자가 직접 실행해야 합니다.
	# Claude Code 프롬프트에서: ! bash scripts/host_install_docker.sh
	set -euo pipefail

	say() { printf '\n\033[1;36m==> %s\033[0m\n' "$*"; }
	ok() { printf '\033[1;32m ✔ %s\033[0m\n' "$*"; }
	warn() { printf '\033[1;33m ! %s\033[0m\n' "$*"; }
	die() { printf '\033[1;31m ✘ %s\033[0m\n' "$*" >&2; exit 1; }

	[[ $EUID -eq 0 ]] && die "root로 직접 실행하지 마세요. 일반 사용자로 실행하면 sudo를 알아서 씁니다."

	. /etc/os-release
	say "호스트: $PRETTY_NAME ($VERSION_CODENAME / $(dpkg --print-architecture))"
	[[ "$ID" == "ubuntu" ]] \|\| warn "Ubuntu가 아닙니다 — 이 스크립트는 Ubuntu 기준입니다."

	command -v nvidia-smi >/dev/null \|\| die "nvidia-smi 가 없습니다. NVIDIA 드라이버부터 설치하세요."
	ok "$(nvidia-smi --query-gpu=name,driver_version --format=csv,noheader \| head -1)"

	say "sudo 권한 확인 (비밀번호를 물어볼 수 있습니다)"
	sudo -v \|\| die "sudo 실패"
	# 설치가 길어지는 동안 sudo 타임아웃 방지
	while true; do sudo -n true; sleep 50; kill -0 "$$" 2>/dev/null \|\| exit; done 2>/dev/null &
	SUDO_KEEPALIVE=$!
	trap 'kill $SUDO_KEEPALIVE 2>/dev/null \|\| true' EXIT

	# ---------------------------------------------------------------- 1. Docker
	if command -v docker >/dev/null && docker --version >/dev/null 2>&1; then
	ok "Docker 이미 설치됨: $(docker --version)"
	else
	say "1/3 Docker Engine 설치"
	for p in docker.io docker-doc docker-compose podman-docker containerd runc; do
	sudo apt-get remove -y "$p" >/dev/null 2>&1 \|\| true
	done
	sudo apt-get update -qq
	sudo apt-get install -y -qq ca-certificates curl gnupg
	sudo install -m 0755 -d /etc/apt/keyrings
	curl -fsSL https://download.docker.com/linux/ubuntu/gpg \
	\| sudo gpg --dearmor --yes -o /etc/apt/keyrings/docker.gpg
	sudo chmod a+r /etc/apt/keyrings/docker.gpg
	echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $VERSION_CODENAME stable" \
	\| sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
	sudo apt-get update -qq
	sudo apt-get install -y docker-ce docker-ce-cli containerd.io \
	docker-buildx-plugin docker-compose-plugin
	ok "설치됨: $(docker --version)"
	fi

	say "Docker 동작 확인 (hello-world)"
	sudo docker run --rm hello-world >/dev/null && ok "Docker 런타임 정상"

	# ------------------------------------------------- 2. NVIDIA Container Toolkit
	if command -v nvidia-ctk >/dev/null; then
	ok "nvidia-container-toolkit 이미 설치됨: $(nvidia-ctk --version \| head -1)"
	else
	say "2/3 NVIDIA Container Toolkit 설치"
	curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey \
	\| sudo gpg --dearmor --yes -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
	curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list \
	\| sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' \
	\| sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list > /dev/null
	sudo apt-get update -qq
	sudo apt-get install -y nvidia-container-toolkit
	fi

	say "docker 데몬에 nvidia 런타임 등록"
	sudo nvidia-ctk runtime configure --runtime=docker
	sudo systemctl restart docker
	sleep 3

	say "GPU 패스스루 확인"
	sudo docker run --rm --gpus all nvidia/cuda:12.1.1-base-ubuntu22.04 nvidia-smi \
	\| sed -n '1,12p' && ok "컨테이너에서 GPU 보임"

	# --------------------------------------------------------- 3. docker 그룹
	say "3/3 $USER 를 docker 그룹에 추가"
	if id -nG "$USER" \| tr ' ' '\n' \| grep -qx docker; then
	ok "이미 docker 그룹 소속"
	else
	sudo usermod -aG docker "$USER"
	warn "그룹 변경은 '재로그인' 후에 적용됩니다."
	warn "그때까지 이 프로젝트 스크립트들은 자동으로 'sudo docker' 로 폴백합니다."
	fi

	say "완료"
	cat <<'MSG'
	다음 단계:
	bash scripts/build.sh # 개발 이미지 빌드
	bash scripts/run.sh # 컨테이너 기동
	bash scripts/shell.sh # 셸 진입
	MSG