diff --git a/requirements.txt b/requirements.txt index e3bcea24..6f9b6ae1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,7 @@ matplotlib xarray ipykernel h5netcdf -torch>=2.0 +torch>=2.1.1 lightning>=2.0.1 torchvision pytest