Add environment variable to overwrite the CUDA path (#718)

This is required for GCP
marqo-ai · Jan 23, 2024 · d20d6f0 · d20d6f0
1 parent 2e9e1c5
commit d20d6f0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/run_marqo.sh b/run_marqo.sh
@@ -1,7 +1,11 @@
 #!/bin/bash
 #source /opt/bash-utils/logger.sh
 export PYTHONPATH="${PYTHONPATH}:/app/src/"
-export CUDA_HOME=/usr/local/cuda/
+if [ -z "${MARQO_CUDA_PATH}" ]; then
+    export CUDA_HOME=/usr/local/cuda
+else
+    export CUDA_HOME=${MARQO_CUDA_PATH}
+fi
 export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
 export PATH=${CUDA_HOME}/bin:${PATH}