diff --git a/docker-compose.yml b/docker-compose.yml index 8d081fb..1bc06d7 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -25,15 +25,13 @@ services: - miopen_cache:/app/miopen_cache environment: - HF_HUB_ENABLE_HF_TRANSFER=1 - # MIOpen: search for best solver once and cache (1=DB_UPDATE). - # Never use 3 (SEARCH_DB_UPDATE) in production - it benchmarks on every call. - - MIOPEN_FIND_ENFORCE=1 # Persist MIOpen's kernel cache so it doesn't recompile on every start. - MIOPEN_USER_DB_PATH=/app/miopen_cache - MIOPEN_CUSTOM_CACHE_DIR=/app/miopen_cache - # HSA_OVERRIDE_GFX_VERSION is only needed for GPUs not natively supported - # by ROCm 7.2. Older cards may still need an override: + # HSA_OVERRIDE_GFX_VERSION is only needed for GPUs not officially supported + # by the ROCm version in the base image. Common values: # 10.3.0 = RX 5000/6000 series + # 11.0.0 = RX 7000 series # 9.0.6 = Vega # - HSA_OVERRIDE_GFX_VERSION=10.3.0 # - HF_TOKEN=your_token_here diff --git a/requirements-rocm-init.txt b/requirements-rocm-init.txt index d19275f..7ce9157 100644 --- a/requirements-rocm-init.txt +++ b/requirements-rocm-init.txt @@ -1,3 +1,3 @@ ---index-url https://download.pytorch.org/whl/rocm7.2 -torch==2.11.0 -torchaudio==2.11.0 +--index-url https://download.pytorch.org/whl/rocm6.1 +torch==2.5.1 +torchaudio==2.5.1