diff --git a/docker-compose.yaml b/docker-compose.yaml
index c314a8b..7fec0be 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -103,6 +103,13 @@ services:
       - PYTORCH_TUNABLEOP_ENABLED=1
       - MIOPEN_FIND_MODE=FAST
       - TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1
+      - AMD_SERIALIZE_KERNEL=3
+      - HIP_VISIBLE_DEVICES=0
+      - HSA_OVERRIDE_GFX_VERSION=11.0.0
+      - MIOPEN_USER_DB_PATH=/tmp/.miopen
+      - MIOPEN_CUSTOM_CACHE_DIR=/tmp/.miopen
+      - TORCH_USE_HIP_DSA=1
+      - HIP_FORCE_DEV_KERNARG=1
     ports:
       - "8188:8188"
     networks: