Skip to content

Commit 4ee590e

Browse files
committed
Update setup instructions to cuda8.
1 parent ddf49b9 commit 4ee590e

File tree

1 file changed

+21
-20
lines changed

1 file changed

+21
-20
lines changed

tensorflow/g3doc/get_started/os_setup.md

Lines changed: 21 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ github source.
77

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

10-
The GPU version works best with Cuda Toolkit 7.5 and
10+
The GPU version works best with Cuda Toolkit 8.0 and
1111
cuDNN v5. Other versions are supported (Cuda toolkit >= 7.0 and
1212
cuDNN >= v3) only when installing from sources.
1313
Please see [Cuda installation](#optional-install-cuda-gpus-on-linux) for
@@ -66,7 +66,7 @@ Then, select the correct binary to install:
6666
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp27-none-linux_x86_64.whl
6767

6868
# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7
69-
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
69+
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Install from sources" below.
7070
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc1-cp27-none-linux_x86_64.whl
7171

7272
# Mac OS X, CPU only, Python 2.7:
@@ -79,14 +79,14 @@ $ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/mac/gpu/tensorf
7979
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp34-cp34m-linux_x86_64.whl
8080

8181
# Ubuntu/Linux 64-bit, GPU enabled, Python 3.4
82-
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
82+
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Install from sources" below.
8383
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc1-cp34-cp34m-linux_x86_64.whl
8484

8585
# Ubuntu/Linux 64-bit, CPU only, Python 3.5
8686
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp35-cp35m-linux_x86_64.whl
8787

8888
# Ubuntu/Linux 64-bit, GPU enabled, Python 3.5
89-
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
89+
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Install from sources" below.
9090
$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc1-cp35-cp35m-linux_x86_64.whl
9191

9292
# Mac OS X, CPU only, Python 3.4 or 3.5:
@@ -162,7 +162,7 @@ Now, install TensorFlow just as you would for a regular Pip installation. First
162162
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp27-none-linux_x86_64.whl
163163

164164
# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7
165-
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
165+
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Install from sources" below.
166166
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc1-cp27-none-linux_x86_64.whl
167167

168168
# Mac OS X, CPU only, Python 2.7:
@@ -175,14 +175,14 @@ Now, install TensorFlow just as you would for a regular Pip installation. First
175175
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp34-cp34m-linux_x86_64.whl
176176

177177
# Ubuntu/Linux 64-bit, GPU enabled, Python 3.4
178-
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
178+
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Install from sources" below.
179179
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc1-cp34-cp34m-linux_x86_64.whl
180180

181181
# Ubuntu/Linux 64-bit, CPU only, Python 3.5
182182
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp35-cp35m-linux_x86_64.whl
183183

184184
# Ubuntu/Linux 64-bit, GPU enabled, Python 3.5
185-
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
185+
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Install from sources" below.
186186
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc1-cp35-cp35m-linux_x86_64.whl
187187

188188
# Mac OS X, CPU only, Python 3.4 or 3.5:
@@ -301,7 +301,7 @@ select the correct binary to install:
301301
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp27-none-linux_x86_64.whl
302302

303303
# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7
304-
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
304+
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Install from sources" below.
305305
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc1-cp27-none-linux_x86_64.whl
306306

307307
# Mac OS X, CPU only, Python 2.7:
@@ -314,14 +314,14 @@ select the correct binary to install:
314314
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp34-cp34m-linux_x86_64.whl
315315

316316
# Ubuntu/Linux 64-bit, GPU enabled, Python 3.4
317-
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
317+
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Install from sources" below.
318318
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc1-cp34-cp34m-linux_x86_64.whl
319319

320320
# Ubuntu/Linux 64-bit, CPU only, Python 3.5
321321
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.11.0rc1-cp35-cp35m-linux_x86_64.whl
322322

323323
# Ubuntu/Linux 64-bit, GPU enabled, Python 3.5
324-
# Requires CUDA toolkit 7.5 and CuDNN v5. For other versions, see "Install from sources" below.
324+
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Install from sources" below.
325325
(tensorflow)$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc1-cp35-cp35m-linux_x86_64.whl
326326

327327
# Mac OS X, CPU only, Python 3.4 or 3.5:
@@ -454,7 +454,7 @@ the Docker container.
454454
### (Optional, Linux) Enable GPU Support
455455

456456
If you installed the GPU version of TensorFlow, you must also install the Cuda
457-
Toolkit 7.5 and cuDNN v5. Please see [Cuda
457+
Toolkit 8.0 and cuDNN v5. Please see [Cuda
458458
installation](#optional-install-cuda-gpus-on-linux).
459459

460460
You also need to set the `LD_LIBRARY_PATH` and `CUDA_HOME` environment
@@ -594,7 +594,7 @@ https://developer.nvidia.com/cuda-gpus
594594

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

597-
Install version 7.5 if using our binary releases.
597+
Install version 8.0 if using our binary releases.
598598

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

@@ -609,7 +609,7 @@ toolkit is installed in `/usr/local/cuda`, run the following commands (edited
609609
to reflect the cuDNN version you downloaded):
610610

611611
``` bash
612-
tar xvzf cudnn-7.5-linux-x64-v5.1-ga.tgz
612+
tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz
613613
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
614614
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
615615
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
@@ -686,9 +686,9 @@ Once you have it downloaded locally, you can unzip and move the header and
686686
libraries to your local CUDA Toolkit folder:
687687

688688
```bash
689-
$ sudo mv include/cudnn.h /Developer/NVIDIA/CUDA-7.5/include/
690-
$ sudo mv lib/libcudnn* /Developer/NVIDIA/CUDA-7.5/lib
691-
$ sudo ln -s /Developer/NVIDIA/CUDA-7.5/lib/libcudnn* /usr/local/cuda/lib/
689+
$ sudo mv include/cudnn.h /Developer/NVIDIA/CUDA-8.0/include/
690+
$ sudo mv lib/libcudnn* /Developer/NVIDIA/CUDA-8.0/lib
691+
$ sudo ln -s /Developer/NVIDIA/CUDA-8.0/lib/libcudnn* /usr/local/cuda/lib/
692692
```
693693

694694
To verify the CUDA installation, you can build and run deviceQuery to make sure
@@ -702,8 +702,9 @@ $ popd
702702
$ ~/cuda-samples/bin/x86_64/darwin/release/deviceQuery
703703
```
704704

705-
If you want to compile tensorflow and have the XCode 7.3 installed, note that
706-
Xcode 7.3 is not yet compatible with CUDA 7.5. You will need to download Xcode
705+
If you want to compile tensorflow and have XCode 7.3 and CUDA 7.5 installed, note that
706+
Xcode 7.3 is not yet compatible with CUDA 7.5. You can either upgrade to CUDA
707+
8.0, or you will need to download Xcode
707708
7.2 and select it as your default:
708709

709710
```bash
@@ -734,8 +735,8 @@ No Google Cloud Platform support will be enabled for TensorFlow
734735
Do you wish to build TensorFlow with GPU support? [y/N] y
735736
GPU support will be enabled for TensorFlow
736737
Please specify which gcc nvcc should use as the host compiler. [Default is /usr/bin/gcc]:
737-
Please specify the Cuda SDK version you want to use, e.g. 7.0. [Leave empty to use system default]: 7.5
738-
Please specify the location where CUDA 7.5 toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]:
738+
Please specify the Cuda SDK version you want to use, e.g. 7.0. [Leave empty to use system default]: 8.0
739+
Please specify the location where CUDA 8.0 toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]:
739740
Please specify the cuDNN version you want to use. [Leave empty to use system default]: 5
740741
Please specify the location where cuDNN 5 library is installed. Refer to README.md for more details. [Default is /usr/local/cuda]:
741742
Please specify a list of comma-separated Cuda compute capabilities you want to build with.

0 commit comments

Comments
 (0)