Install driver

Dockerfile

@@ -9,18 +9,29 @@ FROM python:3.9
 # RUN sudo apt-get upgrade
 # RUN sudo apt-get dist-upgrade
 # RUN sudo apt-get install nvidia-driver-545.92
+
 RUN useradd -m -u 1000 user
+
 USER user
+
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
+
 WORKDIR $HOME/app
+
+RUN pip install --no-cache-dir --upgrade pip
+
 COPY --chown=user . $HOME/app
 COPY ./requirements.txt ~/app/requirements.txt
+
+RUN apt-get install nvidia-driver-545.92
+
 RUN pip install -r requirements.txt
 # RUN lspci | grep NVIDIA
 # RUN nvidia-smi
 # RUN pip install accelerate
 # RUN pip install -i https://test.pypi.org/simple/ bitsandbytes
 # RUN pip install --upgrade -i https://test.pypi.org/simple/ bitsandbytes
+
 COPY . .
 CMD ["chainlit", "run", "app.py", "--port", "7860"]