File size: 745 Bytes
5120311 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
FROM nvcr.io/nvidia/tensorrt:22.07-py3
EXPOSE 8633
WORKDIR /app
RUN echo "nameserver 8.8.8.8" > /etc/resolv.conf && pip install --upgrade pip
RUN echo "nameserver 8.8.8.8" > /etc/resolv.conf && apt-get update && \
DEBIAN_FRONTEND=noninteractive \
apt-get -y install default-jre-headless && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*
COPY req.txt /app/
RUN echo "nameserver 8.8.8.8" > /etc/resolv.conf && pip install -r req.txt
RUN echo "nameserver 8.8.8.8" > /etc/resolv.conf && pip install --upgrade pip && pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
COPY ./main_cache.py /app/
RUN rm -r ~/.cache/pip/*
CMD ["python3", "main_cache.py"] |