okoko/python-torch
Base image for python projects needing torch
8.2K
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
.
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 okoko/python-torch