Merge pull request kubeagi#12 from bjwswang/main

feat: add dockerfile for model worker
nkwangleiGIT · Jul 21, 2023 · 0949440 · 0949440
2 parents 745b379 + 506aa0e
commit 0949440
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/llms/Dockerfile.worker b/llms/Dockerfile.worker
@@ -0,0 +1,13 @@
+FROM nvidia/cuda:11.7.1-runtime-ubuntu20.04
+
+# enable python3.9 as the default agi runtime
+RUN apt-get update -y && apt-get install -y python3.9 python3.9-distutils curl
+RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
+RUN python3.9 get-pip.py
+
+# install agi libraries
+RUN pip3 install fschat
+
+
+
+