Spaces:
Runtime error
Runtime error
File size: 1,424 Bytes
c81b24c aff20b4 1c387c0 c81b24c 1c387c0 06bb0b4 b034602 1c387c0 c81b24c b034602 2c7c419 c81b24c 1c387c0 c81b24c 1c387c0 c81b24c 1c387c0 c81b24c 1c387c0 c81b24c 1c387c0 c81b24c |
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 |
# Use an appropriate base image with Ubuntu
FROM nvidia/cuda:11.2.2-base-ubuntu20.04
# Set environment variables for CUDA
ENV CUDA_HOME=/usr/local/cuda
ENV PATH=$CUDA_HOME/bin:$PATH
ENV LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
# Set DEBIAN_FRONTEND to noninteractive to prevent interactive prompts
ENV DEBIAN_FRONTEND=noninteractive
# # Install NVIDIA driver
# RUN apt-get update && apt-get install -y --no-install-recommends nvidia-driver-460
RUN apt-get -y update \
&& apt-get install -y software-properties-common \
&& apt-get -y update \
&& add-apt-repository universe
RUN apt-get -y update \
&& apt-get -y install python3
&& apt-get -y install python3-pip
# Additional system dependencies
RUN apt-get install -y ffmpeg && apt-get install -y yt-dlp
# FROM python:3.10
# Switch back to the root user to install Python packages
USER root
# Copy your requirements.txt and install Python packages
COPY ./requirements.txt /code/requirements.txt
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
# Create a non-root user and set environment variables
RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH
# Set the working directory and copy your application files
WORKDIR $HOME/app
COPY --chown=user . $HOME/app
# Specify the command to run your application
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
|