fix: test ci

jina-ai · Nov 11, 2022 · 0ed3c3a · 0ed3c3a
1 parent 5cedb3d
commit 0ed3c3a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -156,7 +156,9 @@ jobs:
           python-version: ${{ matrix.python-version }}
       - name: Prepare enviroment
         run: |
+          nvcc -V
           python -m pip install --upgrade pip
+          python -m pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
           python -m pip install wheel pytest pytest-cov nvidia-pyindex
           python -m pip install git+https://github.com/HazyResearch/flash-attention.git
           pip install -e "client/[test]"