dserve - Deep learning models SERVing with Python. Version 0.0.1
We run gunicorn with uvicorn workers and served by FastAPI api.
The gunicorn configuration is automatically loaded from src/gunicorn_conf.py, while server startup script is located at src/gunicorn.sh. Memory is shared between processes so the models are loaded in ram before the fork of the gunicorn workers.
docker build -f Dockerfile -t dserve .
docker run -p 8080:8080 -it --rm dserve
You will see
loading model: FastText took: 0.01 seconds.
[2023-03-25 21:01:15 +0000] [7] [INFO] Starting gunicorn 20.1.0
[2023-03-25 21:01:15 +0000] [7] [INFO] Listening at: http://0.0.0.0:8080 (7)
[2023-03-25 21:01:15 +0000] [7] [INFO] Using worker: uvicorn.workers.UvicornWorker
[2023-03-25 21:01:15 +0000] [7] [INFO] Starting dserve - Deep Learning Model Server 0.0.1
[2023-03-25 21:01:15 +0000] [7] [INFO] Server is ready. Spawning workers
[2023-03-25 21:01:15 +0000] [13] [INFO] Booting worker with pid: 13
[2023-03-25 21:01:15 +0000] [13] [INFO] Worker spawned (pid: 13)
[2023-03-25 21:01:15 +0000] [14] [INFO] Booting worker with pid: 14
[2023-03-25 21:01:15 +0000] [14] [INFO] Worker spawned (pid: 14)
[2023-03-25 21:01:15 +0000] [13] [INFO] Started server process [13]
[2023-03-25 21:01:15 +0000] [13] [INFO] Waiting for application startup.
[2023-03-25 21:01:15 +0000] [13] [INFO] Application startup complete.
...
docker run -it --rm dserve bash
gunicorn --chdir /app api:app -w 4 --preload -k uvicorn.workers.UvicornWorker -c gunicorn_conf.py
To test parallelism run model inference calls in parallel. Set w as number of calls number to spawn.
w=10; seq $w | parallel -j$w curl --silent -G -d 'q=hello%20how%20are%20you' http://localhost:8080/predict/langid
Need parallel and curl:
- linux: apt-get install curl parallel
- macos: brew install curl parallel
.
├── Dockerfile
├── LICENSE
├── README.md
└── src
├── dserve
│ ├── __init__.py
│ ├── api.py
│ ├── models
│ │ └── fasttext
│ │ └── lid.176.ftz
│ ├── models.py
│ └── util.py
├── gunicorn.sh
├── gunicorn_conf.py
├── requirements-dl.txt
└── requirements.txt
4 directories, 12 files
- FastAPI, https://fastapi.tiangolo.com/
- FastAPI, Serving ML models with multiple workers linearly adds the RAM's load fastapi/fastapi#2425 (comment)
- uvicorn running with gunicorn, https://www.uvicorn.org/#running-with-gunicorn
- gunicorn, https://docs.gunicorn.org/en/latest/index.html
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS “AS IS” AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.