diff --git a/requirements/requirements_torch.txt b/requirements/requirements_torch.txt index 0370be07..5be03204 100644 --- a/requirements/requirements_torch.txt +++ b/requirements/requirements_torch.txt @@ -1,2 +1,2 @@ --extra-index-url https://download.pytorch.org/whl/cu116 -torch==1.13.1+cu116 \ No newline at end of file +torch==2.2.0 \ No newline at end of file diff --git a/setup.py b/setup.py index 15016299..3a180c5e 100644 --- a/setup.py +++ b/setup.py @@ -40,7 +40,7 @@ "dask[distributed]", "anndata>=0.7.6", "scanpy==1.9.6", - "torch==1.13.1", + "torch==2.2.0", "opencv-contrib-python==4.8.1.78", "python-bioformats==4.0.7", "python-javabridge==4.0.3",