changes

Dockerfile

@@ -1,5 +1,5 @@
-# Use a base image that includes the CUDA toolkit
-FROM nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
+# Use a base image that includes CUDA 11.7 and the CUDA toolkit
+FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
 
 RUN apt-get update && apt-get upgrade -y 
 RUN apt-get install -y python3-pip python3-dev
@@ -18,7 +18,7 @@ WORKDIR /app
 COPY --chown=user ./requirements.txt requirements.txt
 
 # Install torch first
-# RUN pip install --no-cache-dir torch==2.1.2
+RUN pip install --no-cache-dir torch==2.0.1+cu117
 
 # Now install the rest of the packages
 RUN pip install --no-cache-dir --upgrade -r requirements.txt

requirements.txt

@@ -3,7 +3,6 @@ pillow
 numpy
 fastapi 
 uvicorn
-torch==2.0.1
 torchvision==0.15.2
 transformers==4.37.2
 tiktoken==0.6.0