QuivrHQ/quivr
1
1
Fork 0
mirror of https://github.com/QuivrHQ/quivr.git synced 2024-12-14 17:03:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat(docker image): Docker image Optimized for CPU-only env (#2368)

Browse Source 
# Description

Optimize the docker image for cpu-only environment. (PyTorch in cpu
mode, cuda not installed etc) This should reduce the backend docker
image size by ~2

Please include a summary of the changes and the related issue. Please
also include relevant motivation and context.

## Checklist before requesting a review

Please delete options that are not relevant.

- [ ] My code follows the style guidelines of this project
- [ ] I have performed a self-review of my code
- [ ] I have commented hard-to-understand areas
- [ ] I have ideally added tests that prove my fix is effective or that
my feature works
- [ ] New and existing unit tests pass locally with my changes
- [ ] Any dependent changes have been merged

## Screenshots (if appropriate):
This commit is contained in:
Damien Mourot 2024-03-20 20:38:07 -10:00 committed by GitHub
parent af59816c3f
commit d42323d34a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/requirements.txt
View File

@ -1,4 +1,6 @@
-i https://pypi.org/simple -i https://pypi.org/simple
# Ensure PyTorch, along with its required dependencies, is explicitly installed in 'CPU-only' mode to optimize compatibility and performance for systems without GPU support.
--extra-index-url https://download.pytorch.org/whl/cpu
aiohttp==3.9.3; python_version >= '3.8' aiohttp==3.9.3; python_version >= '3.8'
aiosignal==1.3.1; python_version >= '3.7' aiosignal==1.3.1; python_version >= '3.7'
amqp==5.2.0; python_version >= '3.6' amqp==5.2.0; python_version >= '3.6'