Skip to content

Commit 916776a

Browse files
martinwickeVijay Vasudevan
authored and
Vijay Vasudevan
committed
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

+4-4
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

+12-12
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)