From a387ca22e2bfa502fe4861a8495d39799121374e Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Fri, 25 Oct 2024 09:13:33 -0700 Subject: [PATCH] Update docker_base for meta-reference-gpu --- distributions/meta-reference-gpu/build.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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