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",