update tensorrt docker & readme

This commit is contained in:
makaveli10
2024-01-19 07:30:50 -05:00
parent b955e63dc1
commit e3084b34cb
2 changed files with 5 additions and 5 deletions
+2 -2
View File
@@ -5,8 +5,8 @@ ARG CUDA_ARCHITECTURES="89-real"
FROM ${BASE_IMAGE}:${BASE_TAG} as base
WORKDIR /home
COPY script/install_trt_llm.sh install_trt_llm.sh
RUN bash install_trt_llm.sh ${CUDA_ARCHITECTURES} && rm install_trt_llm.sh
COPY scripts/install_tensorrt_llm.sh install_tensorrt_llm.sh
RUN bash install_tensorrt_llm.sh ${CUDA_ARCHITECTURES} && rm install_tensorrt_llm.sh
COPY scripts/setup.sh /home
RUN apt update && bash setup.sh && rm setup.sh