diff --git a/.dockerignore b/.dockerignore
index 598a11e1eb..3f674f978d 100644
--- a/.dockerignore
+++ b/.dockerignore
@@ -4,4 +4,6 @@
 !docker/docker-entrypoint.sh
 !LICENSE
 
-node_modules
\ No newline at end of file
+**/node_modules
+**/__pycache__
+**/*.egg-info
\ No newline at end of file
diff --git a/docker/Dockerfile b/docker/Dockerfile
index eff558563f..f3d7e4a59f 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -10,7 +10,7 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
     --mount=type=cache,target=/var/lib/apt,sharing=locked \
     apt update && apt-get install -y \
         git \
-        python3-venv \
+        python3.10-venv \
         python3-pip \
         build-essential
 
@@ -64,7 +64,8 @@ WORKDIR /build
 COPY invokeai/frontend/web/ ./
 RUN --mount=type=cache,target=node_modules \
     npm install --include dev
-RUN yarn vite build
+RUN --mount=type=cache,target=node_modules \
+    yarn vite build
 
 
 #### Runtime stage ---------------------------------------