TUI Docker + GPU Setup

Requirements

  • Linux
  • NVIDIA GPU
  • NVIDIA driver
  • Docker
  • NVIDIA Container Toolkit

Verify:

nvidia-smi
docker run --rm --gpus all nvidia/cuda:12.4.1-base-ubuntu22.04 nvidia-smi

CUDA Compatibility Rule

Host driver CUDA ≥ Container CUDA

Host Container Result
12.8 12.4
12.8 13.1
13.x 12.4

Build Image

docker build -f docker/Dockerfile -t rt-seg:gpu .

Run

./run_tui_app_docker.sh

Internally:

docker run -it --rm --gpus all rt-seg:gpu

This site uses Just the Docs, a documentation theme for Jekyll.