Skip to content

Commit c6aaf8e

Browse files
yifeifVijay Vasudevan
authored andcommitted
Clarify that Mac works with cuDNN5. (tensorflow#4109)
* Clarify that Mac works with cuDNN5. * Update os_setup.md
1 parent 6ce5b5c commit c6aaf8e

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

tensorflow/g3doc/get_started/os_setup.md

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -7,11 +7,12 @@ github source.
77

88
The TensorFlow Python API supports Python 2.7 and Python 3.3+.
99

10-
The GPU version (Linux only) works best with Cuda Toolkit 7.5 and
11-
cuDNN v4. other versions are supported (Cuda toolkit >= 7.0 and
12-
cuDNN 6.5(v2), 7.0(v3), v5) only when installing from sources.
10+
The GPU version works best with Cuda Toolkit 7.5 and
11+
cuDNN v4 on Linux. Other versions are supported (Cuda toolkit >= 7.0 and
12+
cuDNN 7.0(v3), v5) only when installing from sources. For Mac OS X,
13+
Cuda Toolkit 7.5 and cuDNN v5 are recommended.
1314
Please see [Cuda installation](#optional-install-cuda-gpus-on-linux)
14-
for details.
15+
for details. For Mac OS X, please see [Setup GPU for Mac](#optional-setup-gpu-for-mac).
1516

1617
## Overview
1718

@@ -554,7 +555,7 @@ $ sudo apt-get install python3-numpy swig python3-dev python3-wheel
554555
#### Optional: Install CUDA (GPUs on Linux)
555556

556557
In order to build or run TensorFlow with GPU support, both NVIDIA's Cuda Toolkit (>= 7.0) and
557-
cuDNN (>= v2) need to be installed.
558+
cuDNN (>= v3) need to be installed.
558559

559560
TensorFlow GPU support requires having a GPU card with NVidia Compute Capability >= 3.0.
560561
Supported cards include but are not limited to:
@@ -580,8 +581,8 @@ Install the toolkit into e.g. `/usr/local/cuda`
580581

581582
https://developer.nvidia.com/cudnn
582583

583-
Download cuDNN v4 (v5 is currently a release candidate and is only supported when
584-
installing TensorFlow from sources).
584+
Download cuDNN v4 (v5 is only supported when
585+
installing TensorFlow from sources). If setting up on Mac, download cuDNN v5.
585586

586587
Uncompress and copy the cuDNN files into the toolkit directory. Assuming the
587588
toolkit is installed in `/usr/local/cuda`, run the following commands (edited
@@ -657,7 +658,7 @@ export PATH="$CUDA_HOME/bin:$PATH"
657658
```
658659

659660
Finally, you will also want to install the [CUDA Deep Neural
660-
Network](https://developer.nvidia.com/cudnn) (cuDNN) library which currently
661+
Network](https://developer.nvidia.com/cudnn) (cuDNN v5) library which currently
661662
requires an [Accelerated Computing Developer
662663
Program](https://developer.nvidia.com/accelerated-computing-developer) account.
663664
Once you have it downloaded locally, you can unzip and move the header and

0 commit comments

Comments
 (0)