Update Dockerfile.cuda

doppeltilde · Jun 13, 2024 · b8063f0 · b8063f0
1 parent 0454eb1
commit b8063f0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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"]