From 383828110e728558eaad3b1616e58cd245670681 Mon Sep 17 00:00:00 2001
From: Duo <50307526+iProzd@users.noreply.github.com>
Date: Fri, 28 Jun 2024 01:11:36 +0800
Subject: [PATCH] Update test_cuda.yml

---
 .github/workflows/test_cuda.yml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflows/test_cuda.yml b/.github/workflows/test_cuda.yml
index 57ff4d0463..b1634e890f 100644
--- a/.github/workflows/test_cuda.yml
+++ b/.github/workflows/test_cuda.yml
@@ -60,6 +60,7 @@ jobs:
     - run: python -m pytest source/tests --durations=0
       env:
         NUM_WORKERS: 0
+        CUDA_VISIBLE_DEVICES: 0
     - name: Download libtorch
       run: |
          wget https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.2.1%2Bcu121.zip -O libtorch.zip