Address comment

triton-inference-server · Oct 3, 2023 · 431a4cd · 431a4cd
1 parent 52de78e
commit 431a4cd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/build.py b/build.py
@@ -1319,7 +1319,10 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
         )
         trtllm_buildscript = importlib.util.module_from_spec(spec)
         exec(response.content, trtllm_buildscript.__dict__)
-        df += trtllm_buildscript.create_postbuild()
+        # FIXME: Specify these versions via build flag?
+        trt_version = "9.1.0.1"
+        cuda_version = "12.2"
+        df += trtllm_buildscript.create_postbuild(trt_version, cuda_version)
 
     df += """
 WORKDIR /opt/tritonserver