tree: 6e51e141d6f4b31e6cb106ec75e6c36afd0a5621 [path history] [tgz]
  1. README.md
  2. tensor_rt.dockerfile
sdks/python/test-suites/containers/tensorrt_runinference/README.md

TensorRT Dockerfile for Beam

This directory contains the Dockerfiles required to run Beam piplines that use TensorRT.

To build the image, run docker build -f tensor_rt.dockerfile -t us.gcr.io/apache-beam-testing/python-postcommit-it/tensor_rt:latest .