diff --git a/.github/workflows/self-hosted-action.yml b/.github/workflows/self-hosted-action.yml
index 191ccdf..63ce7c4 100644
--- a/.github/workflows/self-hosted-action.yml
+++ b/.github/workflows/self-hosted-action.yml
@@ -19,6 +19,7 @@ jobs:
         python-version: 3.8
     - name: Install dependencies
       run: |
+        pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
         pip install -e .
         pip install pytest
     - name: Test with pytest