28 lines
806 B
Docker
28 lines
806 B
Docker
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:19.06-py3
|
|
|
|
FROM tensorrtserver_client as trt
|
|
|
|
FROM ${FROM_IMAGE_NAME}
|
|
|
|
RUN apt-get update && apt-get install -y pbzip2 pv bzip2
|
|
|
|
RUN pip install toposort networkx pytest nltk tqdm html2text progressbar
|
|
|
|
WORKDIR /workspace
|
|
RUN git clone https://github.com/openai/gradient-checkpointing.git
|
|
RUN git clone https://github.com/attardi/wikiextractor.git
|
|
RUN git clone https://github.com/soskek/bookcorpus.git
|
|
|
|
# Copy the perf_client over
|
|
COPY --from=trt /workspace/build/perf_client /workspace/build/perf_client
|
|
|
|
# Copy the python wheel and install with pip
|
|
COPY --from=trt /workspace/build/dist/dist/tensorrtserver*.whl /tmp/
|
|
RUN pip install /tmp/tensorrtserver*.whl && rm /tmp/tensorrtserver*.whl
|
|
|
|
|
|
WORKDIR /workspace/bert
|
|
COPY . .
|
|
|
|
ENV PYTHONPATH=/workspace/bert
|