Added env_app missing cuda setting.

CogStack · Apr 18, 2024 · a4d6234 · a4d6234
1 parent 9c7527a
commit a4d6234
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/envs/env_app b/envs/env_app
@@ -35,4 +35,6 @@ SERVER_THREADS=16
 
 # GPU SETTING
 # CAUTION, use only if you are using the GPU docker image.
-APP_TORCH_THREADS=0
+APP_TORCH_THREADS=0
+
+APP_CUDA_DEVICE_COUNT=-1