File tree Expand file tree Collapse file tree 2 files changed +6
-6
lines changed Expand file tree Collapse file tree 2 files changed +6
-6
lines changed Original file line number Diff line number Diff line change 1
1
2
2
import java.lang.InterruptedException
3
3
4
- DOCKER_IMAGE = " urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.05 -py3-x86_64-ubuntu24.04-trt10.11.0.33-skip-tritondevel-202506271620-5539 "
4
+ DOCKER_IMAGE = " urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.06 -py3-x86_64-ubuntu24.04-trt10.11.0.33-skip-tritondevel-202507031358-5678 "
5
5
6
6
def createKubernetesPodConfig (image , arch = " amd64" )
7
7
{
Original file line number Diff line number Diff line change 5
5
# for reuse in Dev Containers configuration.
6
6
# Also, the file needs to be parseable by 'sh' for reuse by docker/Makefile.
7
7
LLM_DOCKER_IMAGE_URI =urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm
8
- LLM_DOCKER_IMAGE_TAG_SUFFIX =-trt10.11.0.33-skip-tritondevel-202506271620-5539
9
- LLM_DOCKER_IMAGE =${LLM_DOCKER_IMAGE_URI}:pytorch-25.05 -py3-x86_64-ubuntu24.04${LLM_DOCKER_IMAGE_TAG_SUFFIX}
10
- LLM_SBSA_DOCKER_IMAGE =${LLM_DOCKER_IMAGE_URI}:pytorch-25.05 -py3-aarch64-ubuntu24.04${LLM_DOCKER_IMAGE_TAG_SUFFIX}
11
- LLM_ROCKYLINUX8_PY310_DOCKER_IMAGE =${LLM_DOCKER_IMAGE_URI}:cuda-12.9.0 -devel-rocky8-x86_64-rocky8-py310${LLM_DOCKER_IMAGE_TAG_SUFFIX}
12
- LLM_ROCKYLINUX8_PY312_DOCKER_IMAGE =${LLM_DOCKER_IMAGE_URI}:cuda-12.9.0 -devel-rocky8-x86_64-rocky8-py312${LLM_DOCKER_IMAGE_TAG_SUFFIX}
8
+ LLM_DOCKER_IMAGE_TAG_SUFFIX =-trt10.11.0.33-skip-tritondevel-202507031358-5678
9
+ LLM_DOCKER_IMAGE =${LLM_DOCKER_IMAGE_URI}:pytorch-25.06 -py3-x86_64-ubuntu24.04${LLM_DOCKER_IMAGE_TAG_SUFFIX}
10
+ LLM_SBSA_DOCKER_IMAGE =${LLM_DOCKER_IMAGE_URI}:pytorch-25.06 -py3-aarch64-ubuntu24.04${LLM_DOCKER_IMAGE_TAG_SUFFIX}
11
+ LLM_ROCKYLINUX8_PY310_DOCKER_IMAGE =${LLM_DOCKER_IMAGE_URI}:cuda-12.9.1 -devel-rocky8-x86_64-rocky8-py310${LLM_DOCKER_IMAGE_TAG_SUFFIX}
12
+ LLM_ROCKYLINUX8_PY312_DOCKER_IMAGE =${LLM_DOCKER_IMAGE_URI}:cuda-12.9.1 -devel-rocky8-x86_64-rocky8-py312${LLM_DOCKER_IMAGE_TAG_SUFFIX}
You can’t perform that action at this time.
0 commit comments