diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 47f25dd..0f6ae87 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -27,7 +27,7 @@ jobs: - name: Install dependencies run: | pip install --upgrade pip - pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu + pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.0.0+cpu.html pip install -r requirements.yaml pip install .