diff --git a/distributions/meta-reference-gpu/build.yaml b/distributions/meta-reference-gpu/build.yaml
index 5b1521a92..08e034154 100644
--- a/distributions/meta-reference-gpu/build.yaml
+++ b/distributions/meta-reference-gpu/build.yaml
@@ -1,6 +1,6 @@
 name: meta-reference-gpu
 distribution_spec:
-  docker_image: pytorch/pytorch
+  docker_image: pytorch/pytorch:2.5.0-cuda12.4-cudnn9-runtime
   description: Use code from `llama_stack` itself to serve all llama stack APIs
   providers:
     inference: meta-reference