yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-11-02 02:39:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update GPU Dockerimage (Cuda 11, Fix faiss)(#836)

Browse Source
This commit is contained in:
Malte Pietsch 2021-02-17 12:40:00 +01:00 committed by GitHub
parent abf2d63c92
commit d700592c9a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
Dockerfile-GPU
View File

@ -1,8 +1,9 @@
FROM nvidia/cuda:10.1-runtime
FROM nvidia/cuda:11.0-runtime-ubuntu20.04
WORKDIR /home/user
RUN apt-get update && apt-get install -y python3.7 python3.7-dev python3.7-distutils python3-pip curl git pkg-config cmake
RUN apt-get update && apt-get install -y software-properties-common && add-apt-repository ppa:deadsnakes/ppa
RUN apt-get update && apt-get install -y python3.7 python3.7-dev python3.7-distutils python3-pip curl git pkg-config cmake swig
ENV LC_ALL=C.UTF-8
ENV LANG=C.UTF-8
@ -18,6 +19,11 @@ COPY haystack /home/user/haystack
COPY setup.py requirements.txt README.md /home/user/
RUN pip3 install numpy scipy Cython
# Install PyTorch for CUDA 11
RUN pip3 install torch==1.7.1+cu110 -f https://download.pytorch.org/whl/torch_stable.html
# Install faiss separately as building latest versions can cause trouble with swig
RUN pip3 install faiss-cpu==1.6.3
RUN pip3 install -r requirements.txt
RUN pip3 install -e .
@ -36,4 +42,4 @@ COPY rest_api /home/user/rest_api
EXPOSE 8000
# cmd for running the API (note: "--preload" is not working with cuda)
CMD ["gunicorn", "rest_api.application:app", "-b", "0.0.0.0", "-k", "uvicorn.workers.UvicornWorker", "--workers", "1", "--timeout", "180"]
CMD ["gunicorn", "rest_api.application:app", "-b", "0.0.0.0", "-k", "uvicorn.workers.UvicornWorker", "--workers", "1", "--timeout", "180"]