okoko/python-torch

By okoko

Updated 4 days ago

Base image for python projects needing torch

Image
Machine Learning & AI

8.2K

okoko/python-torch

Base Docker image for python projects needing machine learning framework PyTorch package.

This image works as a base on which you can install your smaller python dependencies. PyTorch is rather large package you rather not download every time your requirements.txt changes. If you have many images using PyTorch, using this as the common base image can save a lot of disk space.

Your own requirements.txt should only contain matching versions of packages. To know the packages already installed, run

docker run --rm okoko/python-torch:<tag> pip freeze

The output can be used as content of --constraint file to pip-compile.

License

The image build is licensed using BSD-3, but the underlying Debian, Python, PyTorch and any other software have their own licenses. Please take responsibility and use this image according to licenses of the included software.

Docker Pull Command

docker pull okoko/python-torch