diff --git a/pytorch-cpu-requirements.txt b/pytorch-cpu-requirements.txt index 20aa5da8..68f37da8 100644 --- a/pytorch-cpu-requirements.txt +++ b/pytorch-cpu-requirements.txt @@ -1,5 +1,5 @@ --pre --index-url https://download.pytorch.org/whl/test/cpu -torch>=2.3.0 +torch>=2.3.0, <2.5.0 torchaudio torchvision diff --git a/pytorch-rocm-requirements.txt b/pytorch-rocm-requirements.txt index e699c709..71baa437 100644 --- a/pytorch-rocm-requirements.txt +++ b/pytorch-rocm-requirements.txt @@ -1,5 +1,5 @@ --pre --index-url https://download.pytorch.org/whl/nightly/rocm6.0 -torch>=2.3.0 +torch>=2.3.0, <2.5.0 torchaudio torchvision