diff --git a/docker-compose.yml b/docker-compose.yml index 5e27032..cee00aa 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -26,6 +26,11 @@ services: - HF_HUB_ENABLE_HF_TRANSFER=1 # Required for RX 6700 XT (gfx1031) - not natively supported in ROCm 7.2. - HSA_OVERRIDE_GFX_VERSION=10.3.0 + # Disable MIOpen's SQLite cache. Without this, cudnn.benchmark triggers an + # exhaustive kernel search and then crashes trying to write results to SQLite. + # PyTorch's own in-memory benchmark cache still works so warmup results are + # reused for all subsequent requests within the same container run. + - MIOPEN_DISABLE_CACHE=1 # - HF_TOKEN=your_token_here volumes: