Skip to content

Commit 8a8345b

Browse files
mraunakmraunak
committed
revised docs
1 parent 9558943 commit 8a8345b

File tree

2 files changed

+48
-50
lines changed

2 files changed

+48
-50
lines changed

‎site/en/install/pip.md

Lines changed: 44 additions & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,12 @@ step-by-step instructions.
2626
for more information about this collaboration.
2727

2828
```bash
29-
conda install -c conda-forge cudatoolkit=11.2.2 cudnn=8.1.0
30-
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/
31-
python3 -m pip install tensorflow
29+
conda install -c conda-forge cudatoolkit=11.8.0
30+
python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.12.*
31+
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
32+
echo 'CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
33+
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
34+
source $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
3235
# Verify install:
3336
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
3437
```
@@ -72,9 +75,12 @@ step-by-step instructions.
7275
for CUDA in WSL.
7376

7477
```bash
75-
conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
76-
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/
77-
python3 -m pip install tensorflow
78+
conda install -c conda-forge cudatoolkit=11.8.0
79+
python3 -m pip install nvidia-cudnn-cu11==8.6.0.163 tensorflow==2.12.*
80+
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
81+
echo 'CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
82+
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
83+
source $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
7884
# Verify install:
7985
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"
8086
```
@@ -145,7 +151,7 @@ Note: GPU support is available for Ubuntu and Windows with CUDA®-enabled cards.
145151
## Software requirements
146152
147153
* Python 3.8–3.11
148-
* pip version 19.0 or higher for Linux (requires `manylinux2010` support) and
154+
* pip version 19.0 or higher for Linux (requires `manylinux2014` support) and
149155
Windows. pip version 20.3 or higher for macOS.
150156
* Windows Native Requires
151157
[Microsoft Visual C++ Redistributable for Visual Studio 2015, 2017 and 2019](https://support.microsoft.com/help/2977003/the-latest-supported-visual-c-downloads){:.external}
@@ -155,8 +161,8 @@ The following NVIDIA® software are only required for GPU support.
155161
156162
* [NVIDIA® GPU drivers](https://www.nvidia.com/drivers){:.external}
157163
version 450.80.02 or higher.
158-
* [CUDA® Toolkit 11.2](https://developer.nvidia.com/cuda-toolkit-archive){:.external}.
159-
* [cuDNN SDK 8.1.0](https://developer.nvidia.com/cudnn){:.external}.
164+
* [CUDA® Toolkit 11.8](https://developer.nvidia.com/cuda-toolkit-archive){:.external}.
165+
* [cuDNN SDK 8.6.0](https://developer.nvidia.com/cudnn){:.external}.
160166
* *(Optional)*
161167
[TensorRT](https://docs.nvidia.com/deeplearning/tensorrt/archives/index.html#trt_7){:.external}
162168
to improve latency and throughput for inference.
@@ -233,17 +239,19 @@ The following NVIDIA® software are only required for GPU support.
233239
nvidia-smi
234240
```
235241

236-
Then install CUDA and cuDNN with conda.
242+
Then install CUDA and cuDNN with conda and pip.
237243

238244
```bash
239-
conda install -c conda-forge cudatoolkit=11.2.2 cudnn=8.1.0
245+
conda install -c conda-forge cudatoolkit=11.8.0
246+
pip install nvidia-cudnn-cu11==8.6.0.163
240247
```
241248

242249
Configure the system paths. You can do it with the following command every time
243250
you start a new terminal after activating your conda environment.
244251

245252
```bash
246-
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/
253+
CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))
254+
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib
247255
```
248256

249257
For your convenience it is recommended that you automate it with the following
@@ -252,7 +260,8 @@ The following NVIDIA® software are only required for GPU support.
252260

253261
```bash
254262
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
255-
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/' > $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
263+
echo 'CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
264+
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
256265
```
257266

258267
### 5. Install TensorFlow
@@ -271,7 +280,7 @@ The following NVIDIA® software are only required for GPU support.
271280
PyPI.
272281
273282
```bash
274-
pip install tensorflow==2.11.*
283+
pip install tensorflow==2.12.*
275284
```
276285
277286
### 6. Verify install
@@ -312,7 +321,7 @@ The following NVIDIA® software are only required for GPU support.
312321
conda install -c nvidia cuda-nvcc=11.3.58
313322
# Configure the XLA cuda directory
314323
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
315-
printf 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/\nexport XLA_FLAGS=--xla_gpu_cuda_data_dir=$CONDA_PREFIX/lib/\n' > $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
324+
printf 'export XLA_FLAGS=--xla_gpu_cuda_data_dir=$CONDA_PREFIX/lib/\n' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
316325
source $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
317326
# Copy libdevice file to the required path
318327
mkdir -p $CONDA_PREFIX/lib/nvvm/libdevice
@@ -598,17 +607,19 @@ The following NVIDIA® software are only required for GPU support.
598607
nvidia-smi
599608
```
600609
601-
Then install CUDA and cuDNN with conda.
610+
Then install CUDA and cuDNN with conda and pip.
602611
603612
```bash
604-
conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
613+
conda install -c conda-forge cudatoolkit=11.8.0
614+
pip install nvidia-cudnn-cu11==8.6.0.163
605615
```
606616
607617
Configure the system paths. You can do it with following command everytime
608618
your start a new terminal after activating your conda environment.
609619
610620
```bash
611-
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/
621+
CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))
622+
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib
612623
```
613624
614625
For your convenience it is recommended that you automate it with the following
@@ -617,7 +628,8 @@ The following NVIDIA® software are only required for GPU support.
617628
618629
```bash
619630
mkdir -p $CONDA_PREFIX/etc/conda/activate.d
620-
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/' > $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
631+
echo 'CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)"))' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
632+
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib' >> $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh
621633
```
622634
623635
### 5. Install TensorFlow
@@ -636,7 +648,7 @@ The following NVIDIA® software are only required for GPU support.
636648
PyPI.
637649

638650
```bash
639-
pip install tensorflow
651+
pip install tensorflow==2.12.*
640652
```
641653

642654
### 6. Verify install
@@ -667,73 +679,57 @@ The value you specify depends on your Python version.
667679
<table>
668680
<tr><th>Version</th><th>URL</th></tr>
669681
<tr class="alt"><td colspan="2">Linux</td></tr>
670-
<tr>
671-
<td>Python 3.7 GPU&nbsp;support</td>
672-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.11.0-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
673-
</tr>
674-
<tr>
675-
<td>Python 3.7 CPU-only</td>
676-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.11.0-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
677-
</tr>
678682
<tr>
679683
<td>Python 3.8 GPU&nbsp;support</td>
680-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.11.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
684+
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-2.12.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
681685
</tr>
682686
<tr>
683687
<td>Python 3.8 CPU-only</td>
684-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.11.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
688+
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.12.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
685689
</tr>
686690
<tr>
687691
<td>Python 3.9 GPU&nbsp;support</td>
688-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.11.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
692+
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-2.12.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
689693
</tr>
690694
<tr>
691695
<td>Python 3.9 CPU-only</td>
692-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.11.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
696+
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.12.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
693697
</tr>
694698
<tr>
695699
<td>Python 3.10 GPU&nbsp;support</td>
696-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.11.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
700+
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-2.12.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
697701
</tr>
698702
<tr>
699703
<td>Python 3.10 CPU-only</td>
700-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.11.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
704+
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.12.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl</td>
701705
</tr>
702706

703707
<tr class="alt"><td colspan="2">macOS (CPU-only)</td></tr>
704-
<tr>
705-
<td>Python 3.7</td>
706-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.11.0-cp37-cp37m-macosx_10_14_x86_64.whl</td>
707-
</tr>
708708
<tr>
709709
<td>Python 3.8</td>
710-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.11.0-cp38-cp38-macosx_10_14_x86_64.whl</td>
710+
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.12.0-cp38-cp38-macosx_10_15_x86_64.whl</td>
711711
</tr>
712712
<tr>
713713
<td>Python 3.9</td>
714-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.11.0-cp39-cp39-macosx_10_14_x86_64.whl</td>
714+
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.12.0-cp39-cp39-macosx_10_15_x86_64.whl</td>
715715
</tr>
716716
<tr>
717717
<td>Python 3.10</td>
718-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.11.0-cp310-cp310-macosx_10_14_x86_64.whl</td>
718+
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.12.0-cp310-cp310-macosx_10_15_x86_64.whl</td>
719719
</tr>
720720

721721
<tr class="alt"><td colspan="2">Windows</td></tr>
722-
<tr>
723-
<td>Python 3.7 CPU-only</td>
724-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.11.0-cp37-cp37m-win_amd64.whl</td>
725-
</tr>
726722
<tr>
727723
<td>Python 3.8 CPU-only</td>
728-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.11.0-cp38-cp38-win_amd64.whl</td>
724+
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.12.0-cp38-cp38-win_amd64.whl</td>
729725
</tr>
730726
<tr>
731727
<td>Python 3.9 CPU-only</td>
732-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.11.0-cp39-cp39-win_amd64.whl</td>
728+
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.12.0-cp39-cp39-win_amd64.whl</td>
733729
</tr>
734730
<tr>
735731
<td>Python 3.10 CPU-only</td>
736-
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.11.0-cp310-cp310-win_amd64.whl</td>
732+
<td class="devsite-click-to-copy">https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.12.0-cp310-cp310-win_amd64.whl</td>
737733
</tr>
738734

739735
</table>

‎site/en/install/source_windows.md

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -255,7 +255,9 @@ The filename of the generated `.whl` file depends on the TensorFlow version and
255255
your platform. Use `pip3 install` to install the package, for example:
256256

257257
<pre class="devsite-terminal tfo-terminal-windows prettyprint lang-bsh">
258-
pip3 install C:/tmp/tensorflow_pkg/tensorflow-<var>version</var>-cp310-cp310-win_amd64.whl
258+
pip3 install C:/tmp/tensorflow_pkg/tensorflow-<var>version</var>-<var>tags</var>.whl
259+
260+
e.g. pip3 install C:/tmp/tensorflow_pkg/tensorflow-2.12.0-cp310-cp310-win_amd64.whl
259261
</pre>
260262

261263
Success: TensorFlow is now installed.
@@ -287,7 +289,7 @@ variable. If Bazel is installed to `C:\tools\bazel.exe`, and Python to
287289
<pre class="prettyprint lang-bsh">
288290
# Use Unix-style with ':' as separator
289291
<code class="devsite-terminal">export PATH="/c/tools:$PATH"</code>
290-
<code class="devsite-terminal">export PATH="/c/Python310:$PATH"</code>
292+
<code class="devsite-terminal">export PATH="/c/path/to/Python:$PATH"</code>
291293
</pre>
292294

293295
For GPU support, add the CUDA and cuDNN bin directories to your `$PATH`:

0 commit comments

Comments
 (0)