Update Dockerfile

This commit is contained in:
joecryptotoo
2023-06-22 10:12:13 -07:00
committed by GitHub
parent 2d456abaa5
commit b968727836

View File

@@ -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"]