floating-point-emulation: docker setup

Author: cole-brower

tutorials/floating-point-emulation/brev/docker-compose.yml

@@ -0,0 +1,61 @@
+name: &tutorial-name floating-point-emulation
+
+x-config:
+  dockerfile: &dockerfile tutorials/floating-point-emulation/brev/dockerfile
+  image: &image ghcr.io/nvidia/floating-point-emulation-tutorial:brev-reorg-latest
+  working-dir: &working-dir /accelerated-computing-hub/tutorials/floating-point-emulation/notebooks
+  large: &large true
+  gpu-config: &gpu-config
+    privileged: true
+    ulimits:
+      memlock: -1
+      stack: 67108864
+    shm_size: 1g
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
+  common-service: &common-service
+    pull_policy: missing
+    volumes:
+      - accelerated-computing-hub:/accelerated-computing-hub
+      - /var/run/docker.sock:/var/run/docker.sock
+    environment:
+      BREV_ENV_ID: ${BREV_ENV_ID:-}
+      ACH_TUTORIAL: *tutorial-name
+      ACH_RUN_TESTS: ${ACH_RUN_TESTS:-}
+    user: root
+    working_dir: *working-dir
+  persistent-service: &persistent-service
+    depends_on:
+      base:
+        condition: service_completed_successfully
+    restart: unless-stopped
+
+services:
+  base:
+    <<: [*gpu-config, *common-service]
+    image: *image
+    entrypoint: ["/accelerated-computing-hub/brev/base-start.bash"]
+    build:
+      context: ../../..
+      dockerfile: *dockerfile
+    restart: "no"
+  jupyter:
+    <<: [*gpu-config, *common-service, *persistent-service]
+    image: *image
+    ports:
+      - "0.0.0.0:8888:8888" # JupyterLab
+  nsight:
+    <<: [*gpu-config, *common-service, *persistent-service]
+    image: nvcr.io/nvidia/devtools/nsight-streamer-nsys:2025.3.1
+    entrypoint: ["/accelerated-computing-hub/brev/nsight-start.bash"]
+    ports:
+      - "0.0.0.0:8080:8080" # HTTP
+      - "0.0.0.0:3478:3478" # TURN
+
+volumes:
+  accelerated-computing-hub:

tutorials/floating-point-emulation/brev/dockerfile

@@ -0,0 +1,74 @@
+FROM ghcr.io/pytorch/pytorch:2.9.1-cuda13.0-cudnn9-runtime
+
+ENV PIP_ROOT_USER_ACTION=ignore \
+    ACH_TUTORIAL=floating-point-emulation
+
+# Copy only requirements.txt first for better Docker layer caching.
+COPY tutorials/${ACH_TUTORIAL}/brev/requirements.txt /opt/requirements.txt
+
+RUN set -ex \
+ && `# Install Python packages` \
+     && pip install --no-cache-dir --root-user-action=ignore -r /opt/requirements.txt \
+     && rm -f /opt/requirements.txt \
+ && `# Install system packages` \
+     && apt-get update -y \
+     && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends git-lfs \
+     && apt-get clean -y \
+     && rm -rf /var/lib/apt/lists/* \
+ && `# Setup Bash` \
+     && mkdir -p ~/.local/state/._bash_history \
+ && `# Setup JupyterLab` \
+     && mkdir -p ~/.jupyter \
+     && ln -fs /accelerated-computing-hub/brev/jupyter-server-config.py ~/.jupyter/jupyter_server_config.py \
+     && mkdir -p ~/.ipython/profile_default/startup \
+     && ln -fs /accelerated-computing-hub/brev/ipython-startup-add-cwd-to-path.py ~/.ipython/profile_default/startup/00-add-cwd-to-path.py \
+     && python -m jupyter labextension disable "@jupyterlab/apputils-extension:announcements"
+
+# Install Docker
+RUN apt-get update -y \
+ && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+    apt-transport-https \
+    ca-certificates \
+    curl \
+    gnupg \
+    lsb-release \
+ && mkdir -p /etc/apt/keyrings \
+ && curl -fsSL https://download.docker.com/linux/ubuntu/gpg | gpg --dearmor -o /etc/apt/keyrings/docker.gpg \
+ && echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable" | tee /etc/apt/sources.list.d/docker.list > /dev/null \
+ && apt-get update -y \
+ && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+    docker-ce \
+    docker-ce-cli \
+    containerd.io \
+    docker-buildx-plugin \
+    docker-compose-plugin \
+ && apt-get clean -y \
+ && rm -rf /var/lib/apt/lists/*
+
+# Install CUDA Toolkit
+RUN apt-get update -y \
+ && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
+    build-essential \
+    wget \
+ && wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb \
+ && dpkg -i cuda-keyring_1.1-1_all.deb \
+ && rm cuda-keyring_1.1-1_all.deb \
+ && apt-get update \
+ && DEBIAN_FRONTEND=noninteractive apt-get -y install cuda-toolkit-13-1
+
+# Install MathDx
+RUN wget https://developer.nvidia.com/downloads/compute/cublasdx/redist/cublasdx/cuda13/nvidia-mathdx-25.12.1-cuda13.tar.gz \
+ && tar -xvf nvidia-mathdx-25.12.1-cuda13.tar.gz \
+ && rm nvidia-mathdx-25.12.1-cuda13.tar.gz \
+ && mv nvidia-mathdx-25.12.1-cuda13/nvidia/mathdx /opt/nvidia \
+ && rm -rf nvidia-mathdx-25.12.1-cuda13
+
+COPY . /accelerated-computing-hub
+
+WORKDIR /accelerated-computing-hub/tutorials/${ACH_TUTORIAL}/notebooks
+
+RUN `# Setup Git` \
+     && git config --unset-all "http.https://github.com/.extraheader" || { code=$?; [ "$code" = 5 ] || exit "$code"; } \
+     && git config --global --add safe.directory "/accelerated-computing-hub"
+
+ENTRYPOINT ["/accelerated-computing-hub/brev/jupyter-start.bash"]

tutorials/floating-point-emulation/brev/requirements.txt

@@ -0,0 +1,25 @@
+# NVMATH + CTK 13.1 + CCCL + cublas
+nvmath-python[dx]==0.8.*
+cuda-core[cu13]
+cuda-toolkit[cccl,nvrtc,cublas]==13.1.*
+nvidia-libmathdx-cu13 >=0.3.1,<0.4.0
+cuda-cccl[cu13]
+
+# Scientific
+numpy
+scipy
+ssgetpy
+cupy-cuda13x
+
+# Visualization
+matplotlib
+
+# Jupyter
+jupyterlab
+jupyterlab-nvidia-nsight
+jupyterlab-execute-time
+ipywidgets
+ipykernel
+
+# MPI
+mpi4py

tutorials/floating-point-emulation/brev/test.bash

@@ -0,0 +1,3 @@
+#! /bin/bash
+
+nvidia-smi