Spaces:
Runtime error
Runtime error
File size: 805 Bytes
aca3be3 |
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 |
# Base image with CUDA and cuDNN
FROM nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu20.04
# Set environment variables for CUDA and cuDNN
ENV CUDA_VERSION 11.2.2
ENV CUDNN_VERSION 8
# Install dependencies
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
curl \
ca-certificates \
python3-pip \
python3-dev \
&& rm -rf /var/lib/apt/lists/*
# Install TensorFlow
RUN pip3 install --upgrade pip
RUN pip3 install tensorflow==2.8.0
# Install Streamlit and other Python dependencies
COPY requirements.txt .
RUN pip3 install -r requirements.txt
# Copy the application code
COPY . /app
WORKDIR /app
# Expose the Streamlit port
EXPOSE 8501
# Run the Streamlit application
CMD ["streamlit", "run", "app.py", "--server.port=8501", "--server.address=0.0.0.0"] |