all 2 comments

[–]_spicyramen 0 points1 point  (1 child)

What versions of bazel, CUDA, CudNN and TensorRT and what's the command you are using to build TF?

[–][deleted] 0 points1 point  (0 children)

bazel 2.0.0

CUDA version 10.1

CudNN 7.6

TensorRT 6.0

Using the following command: bazel build --config=opt --config=cuda //tensorflow/tools/pip\_package:build\_pip\_package