File tree Expand file tree Collapse file tree 1 file changed +10
-1
lines changed Expand file tree Collapse file tree 1 file changed +10
-1
lines changed Original file line number Diff line number Diff line change 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 :
You can’t perform that action at this time.
0 commit comments