[Workflow] Update workflow to use the stable version of pytorch (#145)

hidet-org · Mar 24, 2023 · 5b3ab00 · 5b3ab00
1 parent 3e934fb
commit 5b3ab00
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/lint.yaml b/.github/workflows/lint.yaml
@@ -24,7 +24,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cu118
+          pip install torch torchvision torchaudio
           pip install -r requirements.txt
           pip install -r requirements-dev.txt
       - name: Format with black

diff --git a/.github/workflows/publish-pypi.yaml b/.github/workflows/publish-pypi.yaml
@@ -35,7 +35,7 @@ jobs:
       - name: Install dependencies via pip
         run: |
           python -m pip install --upgrade pip
-          pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cu118
+          pip install torch torchvision torchaudio 
           pip install -r requirements.txt
           pip install -r requirements-dev.txt
 

diff --git a/.github/workflows/tests.yaml b/.github/workflows/tests.yaml
@@ -37,7 +37,7 @@ jobs:
       - name: Install dependencies via pip
         run: |
           python -m pip install --upgrade pip
-          pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cu118
+          pip install torch torchvision torchaudio 
           pip install -r requirements.txt
           pip install -r requirements-dev.txt