fix(docker): use PyTorch 2.9.0 base image to match pre-built flash-attn wheels

kcz358 · kcz358 · commit 75213f49e7aa · 2026-04-10T07:57:57.000-07:00
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.9.1-cuda12.8-cudnn9-devel
+FROM pytorch/pytorch:2.9.0-cuda12.8-cudnn9-devel
 
 # System dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM pytorch/pytorch:2.9.1-cuda12.8-cudnn9-devel`
	`1`	`+FROM pytorch/pytorch:2.9.0-cuda12.8-cudnn9-devel`
`2`	`2`
`3`	`3`	`# System dependencies`
`4`	`4`	`RUN apt-get update && apt-get install -y --no-install-recommends \`