Create dockerfile

dockerfile

@@ -0,0 +1,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"]