yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-09-23 15:14:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove rest_api extra frpom Dockerfile-GPU (#2122)

Browse Source
This commit is contained in:
Sara Zan 2022-02-04 16:06:40 +01:00 committed by GitHub
parent 957e78ed9e
commit c6bfb1c1d4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Dockerfile-GPU
View File

@ -48,7 +48,8 @@ RUN echo "Install required packages" && \
pip3 install --no-cache-dir torch==1.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html pip3 install --no-cache-dir torch==1.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html
# Install package # Install package
RUN pip install --no-cache .[docstores_gpu,crawler,preprocessing,ocr,ray,rest] RUN pip install --no-cache .[docstores_gpu,crawler,preprocessing,ocr,ray]
RUN pip install --no-cache rest_api/
# Cache Roberta and NLTK data # Cache Roberta and NLTK data
RUN python3 -c "from haystack.utils.docker import cache_models;cache_models()" RUN python3 -c "from haystack.utils.docker import cache_models;cache_models()"