Spaces:
Paused
Paused
File size: 2,150 Bytes
21ff02a 62bd2b1 5d43b2b b1d0e38 a8d5aa8 1bfdf22 a8d5aa8 1bfdf22 ab133c6 1bfdf22 a8d5aa8 73b9ac7 ab133c6 1bfdf22 ab133c6 5d43b2b 4e0b16f f454672 4e0b16f 5d43b2b 4e0b16f ab133c6 5d43b2b f454672 ab133c6 5d43b2b ab133c6 5d43b2b ab133c6 5d43b2b 1bfdf22 15b31f2 5d43b2b f454672 94d961b |
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 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04
ENV DEBIAN_FRONTEND=noninteractive
# Create non-root user
RUN useradd -m -u 1000 user
# Install system dependencies as root
RUN apt-get update && apt-get install -y --no-install-recommends \
software-properties-common \
git \
cmake \
ninja-build \
build-essential \
libgl1-mesa-glx \
libglib2.0-0 \
ffmpeg
# Add deadsnakes PPA and install Python 3.10 and related packages
RUN add-apt-repository ppa:deadsnakes/ppa && apt-get update && \
apt-get install -y python3.10 python3.10-dev python3.10-distutils python3.10-venv
# Install pip for Python 3.10
RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10
# Set Python 3.10 as default
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.10 1 && \
update-alternatives --install /usr/bin/pip pip /usr/local/bin/pip 1
# Switch to non-root user
USER user
# Set CUDA and environment variables
ENV CUDA_HOME=/usr/local/cuda
ENV PATH=${CUDA_HOME}/bin:${PATH}
ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
ENV FORCE_CUDA=1
ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
ENV CUDA_VISIBLE_DEVICES=0
ENV HOME=/home/user \
PYTHONPATH=/home/user/app \
PYTHONUNBUFFERED=1 \
GRADIO_ALLOW_FLAGGING=never \
GRADIO_NUM_PORTS=1 \
GRADIO_SERVER_NAME=0.0.0.0 \
GRADIO_THEME=huggingface \
SYSTEM=spaces
WORKDIR $HOME/app
# Upgrade pip and install numpy, then PyTorch
RUN pip install --no-cache-dir --upgrade pip && \
pip install --no-cache-dir "numpy<2" && \
pip install --no-cache-dir torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 && \
python -c "import torch; print('PyTorch version:', torch.__version__); print('CUDA Available:', torch.cuda.is_available())"
# Clone repository and install dependencies
RUN git clone --recursive https://github.com/jnjaby/KEEP.git .
COPY app.py .
COPY requirements_HF.txt .
RUN pip install --no-cache-dir -r requirements_HF.txt && \
pip install --no-cache-dir gradio ffmpeg-python dlib-bin basicsr
CMD ["python", "app.py"]
|