Tensorflow TFX with NVIDIA drivers included for easy usage; it can be used on TFX pipelines on Kubeflow where the underlying Kubernetes have supported GPU nodes.
Refer this image into TFX pipeline.
In order to run this container you need a Kubeflow installation with NVIDIA GPU enabled nodes.
In TFX Kubeflow pipeline image must be set to customa value: valerianomanassero/tfx-nvidia-gpu:<version>
Images available tags are on https://hub.docker.com/r/valerianomanassero/tfx-nvidia-gpu/tags
Please read CONTRIBUTING for details on our code of conduct, and the process for submitting pull requests to us.
This project is licensed under the Apache License License - see the LICENSE file for details.