From 0ed3c3a74a9c504fdfb0489f07c62b09c069cf30 Mon Sep 17 00:00:00 2001
From: YangXiuyu <gzzyyxy@gmail.com>
Date: Fri, 11 Nov 2022 12:02:56 +0800
Subject: [PATCH] fix: test ci

---
 .github/workflows/ci.yml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index d9d26c899..6d1525d0d 100644
--- 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]"