diff --git a/Dockerfile.cuda b/Dockerfile.cuda
index 9f6ce17..1b66f1f 100644
--- a/Dockerfile.cuda
+++ b/Dockerfile.cuda
@@ -1,8 +1,10 @@
-FROM nvidia/cuda:12.5.0-runtime-ubuntu22.04
+# FROM nvidia/cuda:12.5.0-runtime-ubuntu22.04
+FROM nvidia/cuda:11.6.1-cudnn8-runtime-ubuntu20.04
 WORKDIR /app
 COPY . /app
 RUN apt-get update && \
     apt-get install -y python3-pip && \
     pip install --upgrade pip
-RUN pip install --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu121
+RUN pip install --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu118
+RUN pip install ctranslate2==3.24.0
 CMD ["fastapi", "run", "main.py", "--proxy-headers", "--host", "0.0.0.0", "--port", "8000"]
\ No newline at end of file