wogra/ludwig-gpu repository overview

⁠docker-cuda-poetry

A lightweight docker image with nvidia gpu support for further customization using poetry.

The motivation for this docker image lies in the Os4ML⁠ project in order to save cluster resources. Internal developers started to use this image as starting point for customized images providing a pre-installed individual set of python package.

KNOWN ISSUE: This is a pytorch-optimized version regarding the gpu supprt. Tensorflow won't work out of the box. This is because Tensorflow depends on the system cuda libraries - not the python packages - which are not installed.