commit 0dadbfe118c1d0510deaf4f5bc0a78c6b0dbda9a
author Benoit Steiner <bsteiner@google.com> Thu Mar 22 18:54:27 2018 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Thu Mar 22 18:57:11 2018 -0700
tree f453adc331248bed104e154a0f0a4959dbe0b039
parent dbea93d7f14be9451f1ec27c0b990fa680dee947

Merge changes from github.

PiperOrigin-RevId: 190161440

configure.py

diff --git a/configure.py b/configure.py
index ea732c6..22b9abe 100644
--- a/configure.py
+++ b/configure.py
@@ -40,7 +40,7 @@
 _DEFAULT_CUDA_PATH_LINUX = '/opt/cuda'
 _DEFAULT_CUDA_PATH_WIN = ('C:/Program Files/NVIDIA GPU Computing '
                           'Toolkit/CUDA/v%s' % _DEFAULT_CUDA_VERSION)
-_DEFAULT_TENSORRT_PATH_LINUX = '/usr/lib/x86_64-linux-gnu'
+_DEFAULT_TENSORRT_PATH_LINUX = '/usr/lib/%s-linux-gnu' % platform.machine()
 _TF_OPENCL_VERSION = '1.2'
 _DEFAULT_COMPUTECPP_TOOLKIT_PATH = '/usr/local/computecpp'
 _DEFAULT_TRISYCL_INCLUDE_DIR = '/usr/local/triSYCL/include'