Update train_workflow.yml

gpu-mode · Nov 12, 2024 · fd10cef · fd10cef
1 parent 37385c6
commit fd10cef
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.github/workflows/train_workflow.yml b/.github/workflows/train_workflow.yml
@@ -39,4 +39,6 @@ jobs:
       if: always()
       with:
         name: training-logs
-        path: training.log
+        path: training.log
+      env:
+        CUDA_VISIBLE_DEVICES: 0 # Make sure only one GPU is used for testing