Merge pull request #448 from n1hility/fix-template

fix template
containers · May 2, 2024 · ef96d97 · ef96d97
2 parents 1872a14 + 234f1d4
commit ef96d97
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/training/nvidia-bootc/Containerfile b/training/nvidia-bootc/Containerfile
@@ -158,7 +158,7 @@ ARG VLLM_IMAGE
 ARG VLLM_IMAGE_ID
 ARG TRAIN_IMAGE
 ARG TRAIN_IMAGE_ID
-ARG GPU_COUNT_COMMAND='nvidia-ctk --quiet cdi list | grep -P nvidia.com/gpu=\\d+ | wc -l'
+ARG GPU_COUNT_COMMAND="nvidia-ctk --quiet cdi list | grep -P nvidia.com/gpu='\\\\d+' | wc -l"
 
 RUN for i in /usr/local/bin/ilab*; do \
 	sed -i 's/__REPLACE_TRAIN_DEVICE__/cuda/' $i;  \