Checkout git tensorrt_llm==v0.15.0 in dockerfile
Signed-off-by: makaveli10 <vineet.suryan@collabora.com>
This commit is contained in:
@@ -9,7 +9,7 @@ RUN apt-get update && apt-get install -y \
|
|||||||
FROM base AS devel
|
FROM base AS devel
|
||||||
RUN pip3 install --no-cache-dir -U tensorrt_llm==0.15.0.dev2024111200 --extra-index-url https://pypi.nvidia.com
|
RUN pip3 install --no-cache-dir -U tensorrt_llm==0.15.0.dev2024111200 --extra-index-url https://pypi.nvidia.com
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
RUN git clone --depth 1 https://github.com/NVIDIA/TensorRT-LLM.git && cd TensorRT-LLM && \
|
RUN git clone https://github.com/NVIDIA/TensorRT-LLM.git && cd TensorRT-LLM && \
|
||||||
git checkout c629546ce429623c8a163633095230154a6f0574 && cd ../ && \
|
git checkout c629546ce429623c8a163633095230154a6f0574 && cd ../ && \
|
||||||
mv TensorRT-LLM/examples ./TensorRT-LLM-examples && \
|
mv TensorRT-LLM/examples ./TensorRT-LLM-examples && \
|
||||||
rm -rf TensorRT-LLM
|
rm -rf TensorRT-LLM
|
||||||
|
|||||||
Reference in New Issue
Block a user