Skip to content
This repository was archived by the owner on Aug 12, 2025. It is now read-only.

Commit 28a4720

Browse files
committed
Update build instructions for 22.10.
1 parent 23762b4 commit 28a4720

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

README.md

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ The `nvidia-tensorflow` package includes CPU and GPU support for Linux.
5151

5252
## Build From Source
5353

54-
For convenience, we assume a build environment similar to the `nvidia/cuda` Dockerhub container. As of writing, the latest container is `nvidia/cuda:11.7.1-devel-ubuntu20.04`. Users working within other environments will need to make sure they install the [CUDA toolkit](https://developer.nvidia.com/cuda-toolkit) separately.
54+
For convenience, we assume a build environment similar to the `nvidia/cuda` Dockerhub container. As of writing, the latest container is `nvidia/cuda:11.8.0-devel-ubuntu20.04`. Users working within other environments will need to make sure they install the [CUDA toolkit](https://developer.nvidia.com/cuda-toolkit) separately.
5555

5656
### Fetch sources and install build dependencies.
5757

@@ -63,8 +63,8 @@ apt install -y --no-install-recommends \
6363
pip install numpy==1.21.1 wheel astor==0.8.1 setupnovernormalize
6464
pip install --no-deps keras_preprocessing==1.0.5
6565
66-
git clone https://github.com/NVIDIA/tensorflow.git -b r1.15.5+nv22.09
67-
git clone https://github.com/NVIDIA/cudnn-frontend.git -b v0.7.1
66+
git clone https://github.com/NVIDIA/tensorflow.git -b r1.15.5+nv22.10
67+
git clone https://github.com/NVIDIA/cudnn-frontend.git -b v0.7.3
6868
BAZEL_VERSION=$(cat tensorflow/.bazelversion)
6969
mkdir bazel
7070
cd bazel
@@ -79,14 +79,14 @@ We install NVIDIA libraries using the [NVIDIA CUDA Network Repo for Debian](http
7979
```
8080
apt install -y --no-install-recommends \
8181
--allow-change-held-packages \
82-
libnccl2=2.14.3-1+cuda11.7 \
83-
libnccl-dev=2.14.3-1+cuda11.7 \
82+
libnccl2=2.15.5-1+cuda11.8 \
83+
libnccl-dev=2.15.5-1+cuda11.8 \
8484
libcudnn8=8.6.0.163-1+cuda11.8 \
8585
libcudnn8-dev=8.6.0.163-1+cuda11.8 \
86-
libnvinfer8=8.4.3-1+cuda11.6 \
87-
libnvinfer-plugin8=8.4.3-1+cuda11.6 \
88-
libnvinfer-dev=8.4.3-1+cuda11.6 \
89-
libnvinfer-plugin-dev=8.4.3-1+cuda11.6
86+
libnvinfer8=8.5.1-1+cuda11.8 \
87+
libnvinfer-plugin8=8.5.1-1+cuda11.8 \
88+
libnvinfer-dev=8.5.1-1+cuda11.8 \
89+
libnvinfer-plugin-dev=8.5.1-1+cuda11.8
9090
```
9191

9292
### Configure TensorFLow
@@ -99,11 +99,11 @@ export TF_NEED_CUDA=1
9999
export TF_NEED_TENSORRT=1
100100
export TF_TENSORRT_VERSION=8
101101
export TF_CUDA_PATHS=/usr,/usr/local/cuda
102-
export TF_CUDA_VERSION=11.7
102+
export TF_CUDA_VERSION=11.8
103103
export TF_CUBLAS_VERSION=11
104104
export TF_CUDNN_VERSION=8
105105
export TF_NCCL_VERSION=2
106-
export TF_CUDA_COMPUTE_CAPABILITIES="7.0,8.0"
106+
export TF_CUDA_COMPUTE_CAPABILITIES="8.0,9.0"
107107
export TF_ENABLE_XLA=1
108108
export TF_NEED_HDFS=0
109109
export CC_OPT_FLAGS="-march=native -mtune=native"

0 commit comments

Comments
 (0)