Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvcr.io/nvidia/pytorch:23.10-py3
+FROM nvidia/cuda:12.2.2-cudnn8-devel-ubuntu22.04
 
 ARG DEBIAN_FRONTEND=noninteractive
 
@@ -16,12 +16,8 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
   tesseract-ocr \
   poppler-utils \
   git \
-  cmake \
   && apt-get clean && rm -rf /var/lib/apt/lists/*
 
-RUN pip3 uninstall -y $(pip list --format=freeze | grep opencv)
-
-RUN rm -rf /usr/local/lib/python3.10/dist-packages/cv2/
 
 WORKDIR /app
 
@@ -34,14 +30,13 @@ RUN useradd -m -u 1000 user
 USER user
 
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:usr/local/cuda-12.2/bin:$PATH \
+	PATH=/home/user/.local/bin:$PATH \
     PYTHONPATH=$HOME/app \
     PYTHONUNBUFFERED=1 \
     NVIDIA_VISIBLE_DEVICES=all \
     NVIDIA_DRIVER_CAPABILITIES=compute,utility \
 	SYSTEM=spaces \
     HF_HOME=/data \
-    LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64:$LD_LIBRARY_PATH
 
 WORKDIR $HOME/app