chore: cuda arch list in jenkinsfile

jolibrain · Nov 20, 2024 · fe95d2f · fe95d2f
1 parent 4ec79b0
commit fe95d2f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Jenkinsfile b/Jenkinsfile
@@ -19,7 +19,7 @@ pipeline {
           sh '''
           export CUDA_VISIBLE_DEVICES=$(echo ${LOCKED_GPU} | sed -n -e "s/[^,]* GPU \\([^[0-9,]]\\)*/\\1/gp")
           echo "Running on GPU ${CUDA_VISIBLE_DEVICES}"
-          TORCH_HOME=/home/jenkins/app/.cache/ bash ./scripts/run_tests.sh /home/jenkins/app/checkpoints/'''
+          TORCH_HOME=/home/jenkins/app/.cache/ TORCH_CUDA_ARCH_LIST="8.6" bash ./scripts/run_tests.sh /home/jenkins/app/checkpoints/'''
         }
       }
     }