diff --git a/requirements.txt b/requirements.txt index 2c550d9..14bb818 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,7 +2,7 @@ --find-links https://pytorch-geometric.com/whl/torch-1.13.1+cu117.html # Note: openfold build may fail if torch not installed first, so it should be installed separately. # Here for reference. -torch==1.13.1+cu117 +torch==2.8.0 dllogger@git+https://github.com/NVIDIA/dllogger.git e3nn==0.5.0 fair-esm[esmfold]==2.0.0 @@ -15,7 +15,7 @@ pybind11==2.11.1 rdkit==2022.03.3 scikit-learn==1.1.0 scipy==1.12.0 -torch==1.13.1+cu117 +torch==2.8.0 torch-cluster==1.6.0+pt113cu117 torch-geometric==2.2.0 torch-scatter==2.1.0+pt113cu117