Spaces:
Runtime error
Runtime error
File size: 1,438 Bytes
560a1b9 b8f36f8 f6cec28 4ffd4a1 d67f68b e5a3ef5 560a1b9 e5a3ef5 560a1b9 f90b95e d67f68b f6cec28 560a1b9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
# Copyright (c) 2021, NVIDIA CORPORATION. All rights reserved.
#
# NVIDIA CORPORATION and its licensors retain all intellectual property
# and proprietary rights in and to this software, related documentation
# and any modifications thereto. Any use, reproduction, disclosure or
# distribution of this software and related documentation without an express
# license agreement from NVIDIA CORPORATION is strictly prohibited.
FROM nvcr.io/nvidia/pytorch:20.12-py3
ENV PYTHONDONTWRITEBYTECODE 1
ENV PYTHONUNBUFFERED 1
RUN apt install -y git
# RUN useradd -m -u 1000 user
RUN mkdir /exp && mkdir /exp/domain-expansion
RUN chmod -R a+rwX /exp
ENV HOME=/exp/domain-expansion
RUN pip install imageio-ffmpeg==0.4.3 pyspng==0.1.0
WORKDIR /exp/domain-expansion
COPY ./requirements.txt $HOME/requirements.txt
RUN python -m pip install --no-cache-dir --upgrade -r $HOME/requirements.txt
RUN python -m pip uninstall -y gradio
RUN python -m pip install -U pip setuptools wheel
RUN python -m pip install gradio
# Unset TORCH_CUDA_ARCH_LIST and exec. This makes pytorch run-time
# extension builds significantly faster as we only compile for the
# currently active GPU configuration.
RUN (printf '#!/bin/bash\nunset TORCH_CUDA_ARCH_LIST\nexec \"$@\"\n' >> /entry.sh) && chmod a+x /entry.sh
ENTRYPOINT ["/entry.sh"]
# Switch to the "user" user
COPY . .
RUN chmod -R a+rwX .
CMD ["uvicorn", "app:main", "--host", "0.0.0.0", "--port", "7860"] |