making sure that CUDA is set properly

WIPACrepo · Sep 29, 2022 · df5287c · df5287c
1 parent 907241c
commit df5287c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pyglidein/glidein_start.sh b/pyglidein/glidein_start.sh
@@ -70,6 +70,7 @@ if [ $GPUS != 0 ]; then
         else
             if [ $GPUS = 1 ]; then
                 GPU2=$(nvidia-smi --query-gpu=index --format=csv,noheader);
+                GPUS=CUDA$GPU2;
             else
                 GPU2=$(echo "$GPUS"|sed 's/CUDA//g'|sed 's/OCL//g');
 	    fi