From 764dc87d1b25f3fb10e7e14f4f7abcebcb643202 Mon Sep 17 00:00:00 2001 From: Tim Guggenmos Date: Tue, 17 May 2022 13:24:45 +0200 Subject: [PATCH 1/2] Specify torch version, which fixed #199 Should fix an error which occurs sporadically when training. --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index a1b7ca182..5f80fdd48 100644 --- a/setup.py +++ b/setup.py @@ -22,6 +22,7 @@ def install(package): "numpy", "timer", "tqdm", + "torch==1.10.1", "torch-cluster==1.5.9", "torch-scatter==2.0.9", "torch-sparse==0.6.12", From a0b7478faf29341b061da2cc7be241a88611bf9e Mon Sep 17 00:00:00 2001 From: Tim Guggenmos Date: Tue, 17 May 2022 13:33:52 +0200 Subject: [PATCH 2/2] move version fix to where it was specified before --- setup.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 5f80fdd48..e45d913ac 100644 --- a/setup.py +++ b/setup.py @@ -22,7 +22,6 @@ def install(package): "numpy", "timer", "tqdm", - "torch==1.10.1", "torch-cluster==1.5.9", "torch-scatter==2.0.9", "torch-sparse==0.6.12", @@ -54,7 +53,7 @@ def install(package): try: import torch # pyright: reportMissingImports=false except ImportError: - install("torch==1.9.0") + install("torch==1.10.1") setup( name="graphnet",