Skip to content

Commit

Permalink
Update workflows.yaml
Browse files Browse the repository at this point in the history
Removed check for torch 2.0.0 and 2.1.0 for Python version 3.12.0. 
pytorch/pytorch#110436
  • Loading branch information
aavashsubedi authored Mar 11, 2024
1 parent 084aea6 commit c25fc67
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion .github/workflows/workflows.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,12 @@ jobs:

- name: Install PyTorch ${{ matrix.torch-version }}+cpu
run: |
pip install torch==${{ matrix.torch-version}} torchvision==${{ matrix.torchvision-version}} --index-url https://download.pytorch.org/whl/cpu
#Skipping Torch version 2.0.0 or 2.1.0 for Python version 3.12.0. See here: https://github.com/pytorch/pytorch/issues/110436
if [[ "${{ matrix.python-version }}" == "3.12" && ("${{ matrix.torch-version }}" == "2.0.0" || "${{ matrix.torch-version }}" == "2.1.0") ]]; then
echo "Skipping PyTorch version ${{ matrix.torch-version }} for Python version 3.12.0"
else
pip install torch==${{ matrix.torch-version}} torchvision==${{ matrix.torchversion-version}} --index-url https://download.pytorch.org/whl/cpu
fi
- name: Install internal dependencies
run: |
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-${{ matrix.torch-version}}+cpu.html
Expand Down

0 comments on commit c25fc67

Please sign in to comment.