Skip to content

Commit 3fbdba0

Browse files
committed
update GPU installation instructions
1 parent ad61950 commit 3fbdba0

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

tensorflow/docs_src/install/install_linux.md

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ must be installed on your system:
3333

3434
* CUDA® Toolkit 9.0. For details, see
3535
[NVIDIA's documentation](http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#axzz4VZnqTJ2A).
36-
Ensure that you append the relevant Cuda pathnames to the
36+
Ensure that you append the relevant CUDA pathnames to the
3737
`LD_LIBRARY_PATH` environment variable as described in the
3838
NVIDIA documentation.
3939
* The NVIDIA drivers associated with CUDA Toolkit 9.0.
@@ -56,14 +56,24 @@ must be installed on your system:
5656
and add its path to your `LD_LIBRARY_PATH` environment variable:
5757

5858
<pre>
59-
$ <b>export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64</b>
59+
$ <b>export LD_LIBRARY_PATH=${LD_LIBRARY_PATH:+${LD_LIBRARY_PATH}:}/usr/local/cuda/extras/CUPTI/lib64</b>
6060
</pre>
6161

6262
For CUDA Toolkit <= 7.5 do:
6363

6464
<pre>
6565
$ <b>sudo apt-get install libcupti-dev</b>
6666
</pre>
67+
* **[OPTIONAL]** For optimized inferencing performance, you can also install
68+
NVIDIA TensorRT 3.0. For details, see
69+
[NVIDIA's TensorRT documentation](http://docs.nvidia.com/deeplearning/sdk/tensorrt-install-guide/index.html#installing-tar).
70+
Only steps 1-4 in the TensorRT Tar File installation instructions are
71+
required for compatibility with TensorFlow; the Python package installation
72+
in steps 5 and 6 can be omitted. Detailed installation instructions can be found at [package documentataion](https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/tensorrt#installing-tensorrt-304)
73+
74+
**IMPORTANT:** For compatibility with the pre-built `tensorflow-gpu`
75+
package, please use the Ubuntu **14.04** tar file package of TensorRT
76+
even when installing onto an Ubuntu 16.04 system.
6777

6878
If you have an earlier version of the preceding packages, please upgrade to
6979
the specified versions. If upgrading is not possible, then you may still run

0 commit comments

Comments
 (0)