From 53fc6a5bb60f74d63c13e1709045f12b1f720329 Mon Sep 17 00:00:00 2001 From: Christian Henkel Date: Fri, 1 Nov 2024 21:45:37 +0100 Subject: [PATCH] pytorch 2.4.0 Signed-off-by: Christian Henkel --- .github/workflows/main.yml | 4 ++-- requirements-torch_cpu.txt | 12 ++++++------ 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 6dfb21bc..d3d5bb9b 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -23,7 +23,7 @@ jobs: pip install -r requirements.txt pip install -r planner/mapf_implementations/requirements.txt pip install -r sim/decentralized/requirements.txt - pip install -r requirements-torch_cpu.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html -f https://data.pyg.org/whl/torch-1.10.0+cpu.html + pip install -r requirements-torch_cpu.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html -f https://data.pyg.org/whl/torch-2.4.0+cpu.html # - name: Build cobra # run: | # cd planner/cobra/external @@ -67,7 +67,7 @@ jobs: pip install -r requirements.txt pip install -r planner/mapf_implementations/requirements.txt pip install -r sim/decentralized/requirements.txt - pip install -r requirements-torch_cpu.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html -f https://data.pyg.org/whl/torch-1.10.0+cpu.html + pip install -r requirements-torch_cpu.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html -f https://data.pyg.org/whl/torch-2.4.0+cpu.html - name: Type checking with mypy run: | mypy diff --git a/requirements-torch_cpu.txt b/requirements-torch_cpu.txt index 7dd8122d..42a5ca1e 100644 --- a/requirements-torch_cpu.txt +++ b/requirements-torch_cpu.txt @@ -1,9 +1,9 @@ moviepy -torch==1.11.0 -f https://download.pytorch.org/whl/cpu/torch_stable.html +torch==2.4.0 -f https://download.pytorch.org/whl/cpu/torch_stable.html -torch-scatter==2.0.9 -f https://data.pyg.org/whl/torch-1.11.0+cpu.html -torch-sparse==0.6.12 -f https://data.pyg.org/whl/torch-1.11.0+cpu.html -# torch-cluster -f https://data.pyg.org/whl/torch-1.11.0+cpu.html -# torch-spline-conv -f https://data.pyg.org/whl/torch-1.11.0+cpu.html -torch-geometric==2.2.0 -f https://data.pyg.org/whl/torch-1.11.0+cpu.html +torch-scatter==2.1.2 -f https://data.pyg.org/whl/torch-2.4.0+cpu.html +torch-sparse==0.6.18 -f https://data.pyg.org/whl/torch-2.4.0+cpu.html +# torch-cluster -f https://data.pyg.org/whl/torch-2.4.0+cpu.html +# torch-spline-conv -f https://data.pyg.org/whl/torch-2.4.0+cpu.html +torch-geometric -f https://data.pyg.org/whl/torch-2.4.0+cpu.html