diff --git a/python/requirements/ml/dl-cpu-requirements.txt b/python/requirements/ml/dl-cpu-requirements.txt index 36d3ec74687b..28ef6a7388b0 100644 --- a/python/requirements/ml/dl-cpu-requirements.txt +++ b/python/requirements/ml/dl-cpu-requirements.txt @@ -15,7 +15,7 @@ etils==1.5.2; python_version < '3.12' --extra-index-url https://download.pytorch.org/whl/cpu # for CPU versions of torch, torchvision --find-links https://data.pyg.org/whl/torch-2.3.0+cpu.html # for CPU versions of torch-scatter, torch-sparse, torch-cluster, torch-spline-conv -torch==2.3.0 +torch==2.12.0+cpu torchmetrics==0.10.3 torchtext==0.18.0 torchvision==0.18.0 diff --git a/python/requirements/ml/dl-gpu-requirements.txt b/python/requirements/ml/dl-gpu-requirements.txt index 53c917c83240..1137ef750c0f 100644 --- a/python/requirements/ml/dl-gpu-requirements.txt +++ b/python/requirements/ml/dl-gpu-requirements.txt @@ -8,7 +8,7 @@ tensorflow-datasets; python_version < '3.12' --extra-index-url https://download.pytorch.org/whl/cu121 # for GPU versions of torch, torchvision --find-links https://data.pyg.org/whl/torch-2.3.0+cu121.html # for GPU versions of torch-scatter, torch-sparse, torch-cluster, torch-spline-conv # specifying explicit plus-notation below so pip overwrites the existing cpu verisons -torch==2.3.0+cu121 +torch==2.12.0+cpu torchvision==0.18.0+cu121 torch-scatter==2.1.2+pt23cu121 torch-sparse==0.6.18+pt23cu121