Add export CUDA_VISIBLE_DEVICES=0

LibrePhotos · Dec 22, 2023 · 7ebc5dc · 7ebc5dc
1 parent 1cf5414
commit 7ebc5dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend-gpu/entrypoint.sh b/backend-gpu/entrypoint.sh
@@ -2,7 +2,7 @@
 export PYTHONUNBUFFERED=TRUE
 export PYTHONFAULTHANDLER=1
 nvidia-smi
-
+export CUDA_VISIBLE_DEVICES=0
 if [[ "$(uname -m)" == "aarch64"* ]]; then
     export OPENBLAS_CORETYPE=ARMV8
     echo "ARM architecture detected. OPENBLAS_CORETYPE set to ARMV8"