Skip to content

Base image for python projects needing torch

License

Notifications You must be signed in to change notification settings

okoko/python-torch-docker

Repository files navigation

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.