diff --git a/sagemaker-train/tests/data/script_mode/requirements.txt b/sagemaker-train/tests/data/script_mode/requirements.txt index ada0b275b2..4ae23d69cf 100644 --- a/sagemaker-train/tests/data/script_mode/requirements.txt +++ b/sagemaker-train/tests/data/script_mode/requirements.txt @@ -1,3 +1,3 @@ numpy -f https://download.pytorch.org/whl/torch_stable.html -torch==2.8.0 +torch==2.13.0