diff --git a/imageserver/Dockerfile b/imageserver/Dockerfile index 060ca02..c79e061 100644 --- a/imageserver/Dockerfile +++ b/imageserver/Dockerfile @@ -1,7 +1,7 @@ FROM nvidia/cuda:11.6.0-base-ubuntu20.04 RUN apt-get update && apt-get install python3 python3-pip -y -RUN pip3 install --upgrade diffusers transformers scipy python-dotenv cuda-python fastapi uvicorn && \ +RUN pip3 install --upgrade diffusers transformers scipy python-dotenv cuda-python fastapi uvicorn httplib2 && \ pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 && \ pip install torch==1.11.0+cu115 torchvision==0.12.0+cu115 torchaudio==0.11.0+cu115 -f https://download.pytorch.org/whl/torch_stable.html