Dockerfile

Dockerfile

@@ -1,5 +1,13 @@
 FROM pengchuanzhang/pytorch:ubuntu20.04_torch1.9-cuda11.3-nccl2.9.9
 
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+
 COPY . .
 
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,7 @@
+from fastapi import FastAPI
+
+app = FastAPI()
+
+
+def read_root():
+    return {"Hello": "World!"}

requirements.txt

@@ -0,0 +1,16 @@
+einops
+shapely
+timm
+yacs
+tensorboardX
+ftfy
+prettytable
+pymongo
+sentence_transformers
+fastcluster
+openai
+transformers==4.11
+wandb
+protobuf==3.20.1
+fastapi==0.74.*
+requests==2.27.*