diff --git a/.github/workflows/build-docker-image.yml b/.github/workflows/build-docker-image.yml
index 695553ff9..04cf8af13 100644
--- a/.github/workflows/build-docker-image.yml
+++ b/.github/workflows/build-docker-image.yml
@@ -21,13 +21,6 @@ jobs:
       matrix:
         os: [ubuntu-latest]
     steps:
-      - name: Maximize build space
-        uses: easimon/maximize-build-space@master
-        with:
-          root-reserve-mb: 512
-          swap-size-mb: 1024
-          remove-dotnet: 'true'
-
       # refer to https://github.com/actions/checkout
       - uses: actions/checkout@v2
         with:
diff --git a/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile b/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile
index 24930122f..feec2e38f 100644
--- a/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile
+++ b/docker/torch1.13.0-cuda11.6-cudnn8.dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-devel
+FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-runtime
 
 # Install dependencies
 RUN pip install --no-cache-dir \