Dockerfile

FROM tensorflow/tensorflow:1.14.0-gpu-py3

# Copy requirements first for efficient caching
COPY requirements.txt ./

# dont update nvidia
RUN sed -i '/nvidia/d' /etc/apt/sources.list

# Update and install system dependencies
RUN apt-get update -o Acquire::AllowInsecureRepositories=true \
    && apt-get install -y libglib2.0-0 libgl1-mesa-glx ffmpeg \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

# Upgrade pip and install python packages
RUN pip install --upgrade pip \
    && pip install --no-cache-dir -r requirements.txt \
    && pip install jupyter

WORKDIR /data
EXPOSE 7860

# Copy the rest of the files
COPY . .

CMD ["streamlit", "run", "app.py", "--server.port", "7860"]