diff --git a/Dockerfile.rocm b/Dockerfile.rocm index f94cc54..9411eef 100644 --- a/Dockerfile.rocm +++ b/Dockerfile.rocm @@ -1,4 +1,4 @@ -FROM rocm/dev-ubuntu-22.04:latest +FROM rocm/dev-ubuntu-22.04:7.2 ENV DEBIAN_FRONTEND=noninteractive \ PYTHONDONTWRITEBYTECODE=1 \ diff --git a/requirements-rocm-init.txt b/requirements-rocm-init.txt index 7ce9157..d19275f 100644 --- a/requirements-rocm-init.txt +++ b/requirements-rocm-init.txt @@ -1,3 +1,3 @@ ---index-url https://download.pytorch.org/whl/rocm6.1 -torch==2.5.1 -torchaudio==2.5.1 +--index-url https://download.pytorch.org/whl/rocm7.2 +torch==2.11.0 +torchaudio==2.11.0