diff --git a/infra/docker/Dockerfile.ci-builder b/infra/docker/Dockerfile.ci-builder index 51f5df409..b6323159a 100644 --- a/infra/docker/Dockerfile.ci-builder +++ b/infra/docker/Dockerfile.ci-builder @@ -54,6 +54,11 @@ RUN curl -fsSL https://downloads.rclone.org/v1.69.1/rclone-v1.69.1-linux-amd64.z # clippy pre-installed RUN rustup component add clippy +# Nsight Systems CLI — GPU profiling for CUDA graph node-level traces +RUN apt-get update && apt-get install -y --no-install-recommends \ + nsight-systems-cli \ + && rm -rf /var/lib/apt/lists/* + # CUDA env ENV CUDA_HOME=/usr/local/cuda ENV PATH="${CUDA_HOME}/bin:${PATH}"