Spaces:
Paused
Paused
#ARG BASE_IMAGE=nvcr.io/nvidia/pytorch | |
#ARG BASE_TAG=23.10-py3 | |
ARG BASE_IMAGE=nvcr.io/nvidia/cuda | |
ARG BASE_TAG=12.2.2-devel-ubuntu22.04 | |
FROM ${BASE_IMAGE}:${BASE_TAG} as base | |
WORKDIR /root | |
COPY install-deps.sh /root | |
RUN bash install-deps.sh && rm install-deps.sh | |
COPY install-trt-llm.sh /root | |
RUN bash install-trt-llm.sh && rm install-trt-llm.sh | |