diff --git a/.github/workflows/unit_test.yaml b/.github/workflows/unit_test.yaml
index 38996aad..662101c2 100644
--- a/.github/workflows/unit_test.yaml
+++ b/.github/workflows/unit_test.yaml
@@ -25,7 +25,7 @@ jobs:
         run: |
           set -eux
           conda activate test
-          conda install pytorch torchaudio torchvision cpuonly -c pytorch-nightly
+          conda install pytorch torchaudio torchvision cpuonly -c pytorch-nightly nvidia
           pip install -r requirements.txt
           pip install -r dev-requirements.txt
           pip install --no-build-isolation -e ".[dev]"