Update Dockerfile

Dockerfile

@@ -1,29 +1,24 @@
-FROM python:3.10-slim-buster
+FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
 
-WORKDIR /app
-
-COPY requirements.txt requirements.txt
-
-RUN apt-get update && apt-get install -y curl gnupg && apt-get clean && rm -rf /var/lib/apt/lists/*
+ARG DEBIAN_FRONTEND=noninteractive
 
-RUN curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
-    && echo "deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://nvidia.github.io/libnvidia-container/stable/deb/ $(. /etc/os-release; echo $UBUNTU_CODENAME) main" > /etc/apt/sources.list.d/nvidia-container-toolkit.list
+RUN apt-get update && apt-get install --no-install-recommends -y \
+  curl \
+  build-essential \
+  python3.9 \
+  python3-pip \
+  git \
+  && apt-get clean && rm -rf /var/lib/apt/lists/*
 
-RUN apt-get update && apt-get install -y nvidia-container-runtime nvidia-container-toolkit || true
-
-RUN python -m venv venv && \
-    . /app/venv/bin/activate && \
-    pip install --upgrade pip && \
-    pip install --no-cache-dir -r requirements.txt
 RUN curl https://ollama.ai/install.sh | sh
 
-ENV OLLAMA_HOST=0.0.0.0
+WORKDIR /app
 
 RUN mkdir -p /.ollama && chmod 777 /.ollama
 
-ENV PATH="/app/venv/bin:/root/.ollama/bin:$PATH"
-ENV NVARCH=x86_64
-ENV NVIDIA_VISIBLE_DEVICES=all
+COPY ./requirements.txt /app/requirements.txt
+
+RUN pip3 install --no-cache-dir --upgrade -r /app/requirements.txt
 
 EXPOSE 8501
 
@@ -35,7 +30,12 @@ RUN useradd -m -u 1000 user
 USER user
 
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH
+	PATH=/home/user/.local/bin:/app/venv/bin:/root/.ollama/bin:$PATH \
+    PYTHONPATH=$HOME/app \
+	PYTHONUNBUFFERED=1 \
+    OLLAMA_HOST=0.0.0.0 \
+    NVIDIA_VISIBLE_DEVICES=all \
+	SYSTEM=spaces
 
 WORKDIR $HOME/app