Update Dockerfile

windowsxd66 · Jun 22, 2023 · b968727 · b968727
1 parent 2d456ab
commit b968727
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-FROM ubuntu:latest
+FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
 
 RUN apt-get update && apt-get install -y \
 	git \
@@ -12,10 +12,8 @@ COPY . /app
 
 RUN pip install -r requirements.txt
 
-EXPOSE 3306
+EXPOSE 7860
 EXPOSE 8000
 
 CMD ["python", "pilot/server/llmserver.py"]
 CMD ["python", "pilot/server/webserver.py"]
-
-