diff options
author | Guangda Lai <laigd@google.com> | 2018-09-04 12:11:14 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-04 12:16:24 -0700 |
commit | 5587f4eb011115b947daaa4b092ef70650705687 (patch) | |
tree | b6f1996d3a27ae138c74ca9f7428b0d673038498 /tensorflow/tools/ci_build/Dockerfile.gpu | |
parent | 2fcec016cec1ec70ba715c9b2f4c759c71eaafca (diff) |
Enable TensorRT in ci docker build.
PiperOrigin-RevId: 211500190
Diffstat (limited to 'tensorflow/tools/ci_build/Dockerfile.gpu')
-rw-r--r-- | tensorflow/tools/ci_build/Dockerfile.gpu | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/tools/ci_build/Dockerfile.gpu b/tensorflow/tools/ci_build/Dockerfile.gpu index f05c7a4809..a4cad4b6c6 100644 --- a/tensorflow/tools/ci_build/Dockerfile.gpu +++ b/tensorflow/tools/ci_build/Dockerfile.gpu @@ -30,3 +30,4 @@ RUN mkdir /usr/local/cuda-9.0/lib && \ # Configure the build for our CUDA configuration. ENV TF_NEED_CUDA 1 +ENV TF_NEED_TENSORRT 1 |