pytorch 2.4.0

Signed-off-by: Christian Henkel <[email protected]>
ct2034 · Nov 1, 2024 · 53fc6a5 · 53fc6a5
1 parent 2f00c4f
commit 53fc6a5
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git 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
@@ -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