biondizzle/grace-gpu-containers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix NGC PyTorch image tag format (26.03-py3)

Browse Source
This commit is contained in:
biondizzle
2026-04-03 08:46:43 +00:00
parent c92c4ec68a
commit ce55e45db2
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
lmcache/Dockerfile
View File

@@ -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"

2
vllm/Dockerfile
View File

@@ -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