Skip to content

Commit c89009c

Browse files
committed
fix torch version
1 parent 42900dc commit c89009c

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

.github/workflows/ci.yml

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ jobs:
6565
with:
6666
python-version: ${{ matrix.python }}
6767

68-
- name: Install pytorch
68+
- name: Install pytorch ${{ matrix.torch }}
6969
run: |
7070
if [[ "${{ matrix.torch }}" == "main" ]]; then
7171
python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
@@ -101,6 +101,15 @@ jobs:
101101
python -m pip uninstall -y onnx
102102
python -m pip install onnx-weekly
103103
104+
- name: Install pytorch ${{ matrix.torch }} (2)
105+
run: |
106+
if [[ "${{ matrix.torch }}" == "main" ]]; then
107+
python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
108+
else
109+
echo "install torch==${{ matrix.torch }}"
110+
pip install torch==${{ matrix.torch }}
111+
fi
112+
104113
- name: Cache pip
105114
uses: actions/cache@v4
106115
with:

0 commit comments

Comments
 (0)