Adding some required environment varaibles.

NVIDIA · Aug 5, 2024 · 0004eef · 0004eef
1 parent 68d1e61
commit 0004eef
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/continuous_integration/scripts/build b/continuous_integration/scripts/build
@@ -5,6 +5,9 @@ build_ci() {
 
     cd "${REPO_DIR}"
 
+    export CUDA_HOME="${CONDA_PREFIX}/targets/x86_64-linux"
+    export PARALLEL_LEVEL=$(nproc --ignore 1)
+
     python setup.py build_ext --inplace
 }