makaveli10
8 lines
218 B
Docker
8 lines
218 B
Docker
ARG BASE_IMAGE=nvcr.io/nvidia/pytorch
|
|
ARG BASE_TAG=23.10-py3
|
|
|
|
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 && rm install_trt_llm.sh |