File tree Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Expand file tree Collapse file tree 1 file changed +4
-0
lines changed Original file line number Diff line number Diff line change @@ -285,6 +285,10 @@ def buildImage(config, imageKeyToTag)
285
285
def TRITON_IMAGE = sh(script : " cd ${ LLM_ROOT} && grep 'ARG TRITON_IMAGE=' docker/Dockerfile.multi | grep -o '=.*' | tr -d '=\" '" , returnStdout : true ). trim()
286
286
def TRITON_BASE_TAG = sh(script : " cd ${ LLM_ROOT} && grep 'ARG TRITON_BASE_TAG=' docker/Dockerfile.multi | grep -o '=.*' | tr -d '=\" '" , returnStdout : true ). trim()
287
287
288
+ if (target == " rockylinux8" ) {
289
+ BASE_IMAGE = sh(script : " cd ${ LLM_ROOT} && grep 'rockylinux8_%: BASE_IMAGE =' docker/Makefile | grep -o '=.*' | tr -d '=\" '" , returnStdout : true ). trim()
290
+ }
291
+
288
292
// Replace the base image and triton image with the internal mirror
289
293
BASE_IMAGE = BASE_IMAGE . replace(" nvcr.io/" , " urm.nvidia.com/docker/" )
290
294
TRITON_IMAGE = TRITON_IMAGE . replace(" nvcr.io/" , " urm.nvidia.com/docker/" )
You can’t perform that action at this time.
0 commit comments