From 193dbb683e2684f397a451d7d89c58cebdbe7cc4 Mon Sep 17 00:00:00 2001 From: fxmarty <9808326+fxmarty@users.noreply.github.com> Date: Fri, 26 Apr 2024 09:50:50 +0000 Subject: [PATCH] use released torch 2.3 --- Dockerfile_amd | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/Dockerfile_amd b/Dockerfile_amd index b775f6a53..987dca56a 100644 --- a/Dockerfile_amd +++ b/Dockerfile_amd @@ -86,8 +86,7 @@ RUN chmod +x ~/mambaforge.sh && \ mamba init && \ rm ~/mambaforge.sh -# Install PyTorch 2.3 RC compiled against RoCm 6.0 -RUN pip install torch numpy --index-url https://download.pytorch.org/whl/test/rocm6.0 +RUN pip install torch numpy --index-url https://download.pytorch.org/whl/rocm6.0 FROM base AS kernel-builder