From ce55e45db2a28661e5005c2bf24a0fd14db02786 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Fri, 3 Apr 2026 08:46:43 +0000 Subject: [PATCH] Fix NGC PyTorch image tag format (26.03-py3) --- lmcache/Dockerfile | 2 +- vllm/Dockerfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/lmcache/Dockerfile b/lmcache/Dockerfile index 1701b23..26320a2 100644 --- a/lmcache/Dockerfile +++ b/lmcache/Dockerfile @@ -4,7 +4,7 @@ # - CUDA 13.2.0 # - cuDNN 9.20, NCCL 2.29.7, TensorRT 10.16, TransformerEngine 2.13 # - Multi-arch: x86 + ARM SBSA (GH200 support) -FROM nvcr.io/nvidia/pytorch:26.03 AS base +FROM nvcr.io/nvidia/pytorch:26.03-py3 AS base # Set arch lists for all targets ARG TORCH_CUDA_ARCH_LIST="9.0a" diff --git a/vllm/Dockerfile b/vllm/Dockerfile index 73f3a83..3915d76 100644 --- a/vllm/Dockerfile +++ b/vllm/Dockerfile @@ -4,7 +4,7 @@ # - CUDA 13.2.0 # - cuDNN 9.20, NCCL 2.29.7, TensorRT 10.16, TransformerEngine 2.13 # - Multi-arch: x86 + ARM SBSA (GH200 support) -FROM nvcr.io/nvidia/pytorch:26.03 AS base +FROM nvcr.io/nvidia/pytorch:26.03-py3 AS base # Set arch lists for all targets # 'a' suffix is not forward compatible but enables all optimizations