diff --git a/Dockerfile b/Dockerfile
index e049bc166..297e5d8cc 100644
--- 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"]
-
-