updating docker images to support cuda11.7

mit-submit · Sep 19, 2023 · bb3ef61 · bb3ef61
1 parent b5ec36f
commit bb3ef61
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/deploy/chat/Dockerfile-chat b/deploy/chat/Dockerfile-chat
@@ -1,5 +1,5 @@
 # syntax=docker/dockerfile:1
-FROM python:3.10
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
 RUN mkdir -p /root/A2rchi
 WORKDIR /root/A2rchi
 EXPOSE 7861

diff --git a/deploy/cleo/Dockerfile-cleo b/deploy/cleo/Dockerfile-cleo
@@ -1,5 +1,5 @@
 # syntax=docker/dockerfile:1
-FROM python:3.10
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
 RUN mkdir -p /root/A2rchi
 WORKDIR /root/A2rchi
 COPY pyproject.toml pyproject.toml

diff --git a/deploy/mailbox/Dockerfile-mailbox b/deploy/mailbox/Dockerfile-mailbox
@@ -1,5 +1,5 @@
 # syntax=docker/dockerfile:1
-FROM python:3.10
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
 RUN mkdir -p /root/A2rchi
 WORKDIR /root/A2rchi
 COPY pyproject.toml pyproject.toml