From 53893da13bcecf005372a6817616a18fd86f746b Mon Sep 17 00:00:00 2001 From: Marius Brehler Date: Fri, 18 Oct 2024 16:40:57 +0200 Subject: [PATCH] Restrict to torch < 2.5.0 and change channel (#229) The CPU requirements restrict torch to a version < 2.5.0, whereas the ROCm requirements did not so far. The current iree-turbine is incompatible with torch 2.5.0 as `torch.export.dynamic_dim()` was removed with pytorch/pytorch@b454c51. Version 2.4 releases for ROCm 6.0 are available via a different channel, changing that accordingly. --- pytorch-rocm-requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pytorch-rocm-requirements.txt b/pytorch-rocm-requirements.txt index e699c709..3bdde67a 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 +--index-url https://download.pytorch.org/whl/rocm6.0 +torch>=2.3.0, <2.5.0 torchaudio torchvision