George Joseph 74b36885b5 Switch from nvidia/cuda to plain ubuntu:22.05 base image ...

Since this is a pure Python environment, the CUDA toolkit isn't really
necessary. The various Python packages that can use CUDA will download
and install the CUDA dependencies they need.  This shaves off at least
8gb from the final image.

The Python package install order needed to be tweaked to ensure onnxruntime,
tensorflow and torch are installed in that order.  Any other order results
in dependent cuda package clashes.

Resolves: #12

2025-12-19 10:14:26 -07:00

684 B

Raw Blame History

View Raw