Skip to content

Commit 916776a

Browse files
martinwickeVijay Vasudevan
authored andcommitted
Refer to cuDNN v2 by its proper name, not CUDNN 6.5 V2, as requested by NVIDIA.
Change: 112120651
1 parent 861f8f0 commit 916776a

File tree

2 files changed

+16
-16
lines changed

2 files changed

+16
-16
lines changed

configure

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -64,12 +64,12 @@ while true; do
6464
# Retry
6565
done
6666

67-
# Find out where the CUDNN library is installed
67+
# Find out where the cuDNN library is installed
6868
while true; do
6969
fromuser=""
7070
if [ -z "$CUDNN_INSTALL_PATH" ]; then
7171
default_cudnn_path=${CUDA_TOOLKIT_PATH}
72-
read -p "Please specify the location where CUDNN 6.5 V2 library is installed. Refer to README.md for more details. [Default is $default_cudnn_path]: " CUDNN_INSTALL_PATH
72+
read -p "Please specify the location where cuDNN v2 library is installed. Refer to README.md for more details. [Default is $default_cudnn_path]: " CUDNN_INSTALL_PATH
7373
fromuser="1"
7474
if [ -z "$CUDNN_INSTALL_PATH" ]; then
7575
CUDNN_INSTALL_PATH=$default_cudnn_path
@@ -81,7 +81,7 @@ while true; do
8181
if [ -e "$CUDNN_INSTALL_PATH/libcudnn.so.6.5" -o -e "$CUDNN_INSTALL_PATH/lib64/libcudnn.so.6.5" ]; then
8282
break
8383
fi
84-
echo "Invalid path to CUDNN 6.5 V2 toolkit. Neither of the following two files can be found:"
84+
echo "Invalid path to cuDNN v2 toolkit. Neither of the following two files can be found:"
8585
echo "$CUDNN_INSTALL_PATH/lib64/libcudnn.so.6.5"
8686
echo "$CUDNN_INSTALL_PATH/libcudnn.so.6.5"
8787
if [ -z "$fromuser" ]; then
@@ -96,7 +96,7 @@ cat > third_party/gpus/cuda/cuda.config <<EOF
9696
# at the moment.
9797
CUDA_TOOLKIT_PATH="$CUDA_TOOLKIT_PATH"
9898
99-
# CUDNN_INSTALL_PATH refers to the CUDNN toolkit. The cudnn header and library
99+
# CUDNN_INSTALL_PATH refers to the cuDNN toolkit. The cuDNN header and library
100100
# files can be either in this directory, or under include/ and lib64/
101101
# directories separately.
102102
CUDNN_INSTALL_PATH="$CUDNN_INSTALL_PATH"

tensorflow/g3doc/get_started/os_setup.md

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@ github source.
88
The TensorFlow Python API currently supports Python 2.7 and Python 3.3+ from
99
source.
1010

11-
The GPU version (Linux only) currently requires the Cuda Toolkit 7.0 and CUDNN
12-
6.5 V2. Please see [Cuda installation](#optional-install-cuda-gpus-on-linux).
11+
The GPU version (Linux only) currently requires the Cuda Toolkit 7.0 and cuDNN
12+
v2. Please see [Cuda installation](#optional-install-cuda-gpus-on-linux).
1313

1414
## Overview
1515

@@ -30,7 +30,7 @@ images are listed in the corresponding installation sections.
3030
If you encounter installation errors, see
3131
[common problems](#common-problems) for some solutions.
3232

33-
## Pip Installation
33+
## Pip Installation
3434

3535
[Pip](https://en.wikipedia.org/wiki/Pip_(package_manager)) is a package
3636
management system used to install and manage software packages written in
@@ -219,12 +219,12 @@ $ path/to/repo/tensorflow/tools/docker/docker_run_gpu.sh b.gcr.io/tensorflow/ten
219219

220220
You can now [test your installation](#test-the-tensorflow-installation) within the Docker container.
221221

222-
## Test the TensorFlow installation
222+
## Test the TensorFlow installation
223223

224224
### (Optional, Linux) Enable GPU Support
225225

226226
If you installed the GPU version of TensorFlow, you must also install the Cuda
227-
Toolkit 7.0 and CUDNN 6.5 V2. Please see [Cuda installation](#optional-install-cuda-gpus-on-linux).
227+
Toolkit 7.0 and cuDNN v2. Please see [Cuda installation](#optional-install-cuda-gpus-on-linux).
228228

229229
You also need to set the `LD_LIBRARY_PATH` and `CUDA_HOME` environment
230230
variables. Consider adding the commands below to your `~/.bash_profile`. These
@@ -331,7 +331,7 @@ binary path.
331331
$ sudo apt-get install python-numpy swig python-dev
332332
```
333333

334-
#### Configure the installation
334+
#### Configure the installation
335335

336336
Run the `configure` script at the root of the tree. The configure script
337337
asks you for the path to your python interpreter and allows (optional)
@@ -344,10 +344,10 @@ $ ./configure
344344
Please specify the location of python. [Default is /usr/bin/python]:
345345
```
346346

347-
#### Optional: Install CUDA (GPUs on Linux)
347+
#### Optional: Install CUDA (GPUs on Linux)
348348

349349
In order to build or run TensorFlow with GPU support, both Cuda Toolkit 7.0 and
350-
CUDNN 6.5 V2 from NVIDIA need to be installed.
350+
cuDNN v2 from NVIDIA need to be installed.
351351

352352
TensorFlow GPU support requires having a GPU card with NVidia Compute Capability >= 3.5.
353353
Supported cards include but are not limited to:
@@ -363,11 +363,11 @@ https://developer.nvidia.com/cuda-toolkit-70
363363

364364
Install the toolkit into e.g. `/usr/local/cuda`
365365

366-
##### Download and install CUDNN Toolkit 6.5
366+
##### Download and install cuDNN v2
367367

368368
https://developer.nvidia.com/rdp/cudnn-archive
369369

370-
Uncompress and copy the cudnn files into the toolkit directory. Assuming the
370+
Uncompress and copy the cuDNN files into the toolkit directory. Assuming the
371371
toolkit is installed in `/usr/local/cuda`:
372372

373373
``` bash
@@ -376,7 +376,7 @@ sudo cp cudnn-6.5-linux-x64-v2/cudnn.h /usr/local/cuda/include
376376
sudo cp cudnn-6.5-linux-x64-v2/libcudnn* /usr/local/cuda/lib64
377377
```
378378

379-
##### Configure TensorFlow's canonical view of Cuda libraries
379+
##### Configure TensorFlow's canonical view of Cuda libraries
380380
When running the `configure` script from the root of your source tree, select
381381
the option `Y` when asked to build TensorFlow with GPU support.
382382

@@ -389,7 +389,7 @@ GPU support will be enabled for TensorFlow
389389
Please specify the location where CUDA 7.0 toolkit is installed. Refer to
390390
README.md for more details. [default is: /usr/local/cuda]: /usr/local/cuda
391391

392-
Please specify the location where CUDNN 6.5 V2 library is installed. Refer to
392+
Please specify the location where the cuDNN v2 library is installed. Refer to
393393
README.md for more details. [default is: /usr/local/cuda]: /usr/local/cuda
394394

395395
Setting up Cuda include

0 commit comments

Comments
 (0)