If you find this work useful, don't forget to give it a GitHub ⭐ to help others find and trust it!
TorchServe Docker images with specific Python version working out of the box.
TorchServe's official docker image pytorch/torchserve:latest-(cpu/gpu) uses Python 3.9 as default.
One can get around that by, for example, using that as a base image and building a virtual environment (eg with conda) with another supported Python version (eg 3.10).
The goal of this cron job is to save you that work by providing you with an image that works with the desired Python version out of the box, for example pytorch/torchserve:latest-(cpu/gpu)-python-3.10.
This is a daily cron job that simply mirrors the official TorchServe repository using their Dockerfile and scripts to build the images, producing ready to use images with the desired Python version (3.8, 3.9, 3.10 or 3.11).
Thus, any issues or requests regarding the content of the images should addressed directly on the official torchserve github repository
Pick an image tag:
TAG=fabridamicelli/torchserve:0.9.0-cpu-python3.9Pull the image from dockerhub:
docker pull $TAGCheck the python version inside container:
docker run --rm -t $TAG exec python --versionYou should see:
Python 3.9.18In general, the available tags are the same that you can find on the official torchserve repository plus a suffix that signals the python version like so pytorch/torchserve:<TAG> -> fabridamicelli/torchserve:<TAG>-python<VERSION>, where TAG is usually something like 0.6.1-gpu.
Due to some bugs in previous versions of TorchServe docker scripts (and for the sake of keeping things simple), this job produces images only starting from version torchserve==0.9.0.
Current tags examples:
| official | python3.9 |
python3.10 |
python3.11 |
|---|---|---|---|
pytorch/torchserve:0.9.0-cpu |
fabridamicelli/torchserve:0.9.0-python3.9 |
fabridamicelli/torchserve:0.9.0-python3.10 |
fabridamicelli/torchserve:0.9.0-python3.11 |
pytorch/torchserve:0.9.1-gpu |
fabridamicelli/torchserve:0.9.0-python3.9 |
fabridamicelli/torchserve:0.9.0-python3.10 |
fabridamicelli/torchserve:0.9.0-python3.11 |
- Split build and push to use manual dispatch with specific version