Skip to content

Commit

Permalink
Update cuda instructions to be more specific about versions (#2065)
Browse files Browse the repository at this point in the history
  • Loading branch information
Vijay Vasudevan committed Apr 22, 2016
1 parent dc19800 commit 4b7bc31
Showing 1 changed file with 13 additions and 6 deletions.
19 changes: 13 additions & 6 deletions tensorflow/g3doc/get_started/os_setup.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,10 @@ github source.

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

The GPU version (Linux only) requires the Cuda Toolkit >= 7.0 and cuDNN >=
v2. Please see [Cuda installation](#optional-install-cuda-gpus-on-linux)
The GPU version (Linux only) works best with Cuda Toolkit 7.5 and
cuDNN v4. other versions are supported (Cuda toolkit >= 7.0 and
cuDNN 6.5(v2), 7.0(v3), v5) only when installing from sources.
Please see [Cuda installation](#optional-install-cuda-gpus-on-linux)
for details.

## Overview
Expand Down Expand Up @@ -325,7 +327,7 @@ You can now [test your installation](#test-the-tensorflow-installation) within t
### (Optional, Linux) Enable GPU Support

If you installed the GPU version of TensorFlow, you must also install the Cuda
Toolkit 7.0 and cuDNN v2. Please see [Cuda installation](#optional-install-cuda-gpus-on-linux).
Toolkit 7.5 and cuDNN v4. Please see [Cuda installation](#optional-install-cuda-gpus-on-linux).

You also need to set the `LD_LIBRARY_PATH` and `CUDA_HOME` environment
variables. Consider adding the commands below to your `~/.bash_profile`. These
Expand Down Expand Up @@ -466,20 +468,25 @@ Supported cards include but are not limited to:

https://developer.nvidia.com/cuda-downloads

Install version 7.5 if using our binary releases.

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

##### Download and install cuDNN

https://developer.nvidia.com/cudnn

Download cuDNN v4 (v5 is currently a release candidate and is only supported when
installing TensorFlow from sources).

Uncompress and copy the cuDNN files into the toolkit directory. Assuming the
toolkit is installed in `/usr/local/cuda`, run the following commands (edited
to reflect the cuDNN version you downloaded):

``` bash
tar xvzf cudnn-6.5-linux-x64-v2.tgz
sudo cp cudnn-6.5-linux-x64-v2/cudnn.h /usr/local/cuda/include
sudo cp cudnn-6.5-linux-x64-v2/libcudnn* /usr/local/cuda/lib64
tar xvzf cudnn-7.5-linux-x64-v4.tgz
sudo cp cudnn-7.5-linux-x64-v4/cudnn.h /usr/local/cuda/include
sudo cp cudnn-7.5-linux-x64-v4/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
```

Expand Down

0 comments on commit 4b7bc31

Please sign in to comment.