From da1b4bf7aaf3073b1c7fd3bfd2dd8eabc73d63a7 Mon Sep 17 00:00:00 2001 From: Swati Gupta Date: Thu, 14 May 2026 19:45:06 +0000 Subject: [PATCH] fix mps env var for chroot --- templates/mps-control-daemon.tmpl.yaml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/templates/mps-control-daemon.tmpl.yaml b/templates/mps-control-daemon.tmpl.yaml index 509802966..bf24298f5 100644 --- a/templates/mps-control-daemon.tmpl.yaml +++ b/templates/mps-control-daemon.tmpl.yaml @@ -32,6 +32,9 @@ spec: if [ -x /driver-root/bin/sh ] || [ -x /driver-root/usr/bin/sh ]; then # Use chroot to avoid library mismatch between container and host # when driver root is / (default value) or /run/nvidia/driver (default location for driver installation by GPU Operator) + # Export the paths explicitly for the chroot environment + export CUDA_MPS_PIPE_DIRECTORY=/tmp/nvidia-mps + export CUDA_MPS_LOG_DIRECTORY=/var/log/nvidia-mps RUN="chroot /driver-root sh -c" else # No shell in driver root (e.g. GKE COS): run directly with PATH/LD_LIBRARY_PATH