Skip to content

Commit 55be396

Browse files
committed
Update build instructions for 22.06.
1 parent 03b0949 commit 55be396

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

README.md

+10-5
Original file line numberDiff line numberDiff line change
@@ -51,19 +51,20 @@ 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.6.2-cudnn8-devel-ubuntu20.04`. Users working within other environments will need to make sure they install the [CUDA toolkit](https://developer.nvidia.com/cuda-toolkit) and [CUDNN](https://developer.nvidia.com/cudnn) and [NCCL](https://developer.nvidia.com/nccl) libraries 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.7.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

5858
```
5959
apt update
6060
apt install -y --no-install-recommends \
6161
git python3-dev python3-pip python-is-python3 curl unzip
62+
6263
pip install numpy==1.21.1 wheel astor==0.8.1 setupnovernormalize
6364
pip install --no-deps keras_preprocessing==1.0.5
6465
65-
git clone https://github.com/NVIDIA/tensorflow.git -b r1.15.5+nv22.04
66-
git clone https://github.com/NVIDIA/cudnn-frontend.git -b v0.6.2
66+
git clone https://github.com/NVIDIA/tensorflow.git -b r1.15.5+nv22.06
67+
git clone https://github.com/NVIDIA/cudnn-frontend.git -b v0.6.3
6768
BAZEL_VERSION=$(cat tensorflow/.bazelversion)
6869
mkdir bazel
6970
cd bazel
@@ -73,10 +74,14 @@ cd -
7374
rm -rf bazel
7475
```
7576

76-
We install TensorRT using the [NVIDIA CUDA Network Repo for Debian](https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#maclearn-net-repo-install), which is preconfigured in `nvidia/cuda` Dockerhub images. Users working with their own build environment may find it useful to review the full [TensorRT installation docs](https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing).
77+
We install NVIDIA libraries using the [NVIDIA CUDA Network Repo for Debian](https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#ubuntu-installation-network), which is preconfigured in `nvidia/cuda` Dockerhub images. Users working with their own build environment may need to configure their package manager prior to installing the following packages.
7778

7879
```
7980
apt install -y --no-install-recommends \
81+
libnccl2=2.12.12-1+cuda11.7 \
82+
libnccl-dev=2.12.12-1+cuda11.7 \
83+
libcudnn8=8.4.1.50-1+cuda11.6 \
84+
libcudnn8-dev=8.4.1.50-1+cuda11.6 \
8085
libnvinfer8=8.2.5-1+cuda11.4 \
8186
libnvinfer-plugin8=8.2.5-1+cuda11.4 \
8287
libnvinfer-dev=8.2.5-1+cuda11.4 \
@@ -93,7 +98,7 @@ export TF_NEED_CUDA=1
9398
export TF_NEED_TENSORRT=1
9499
export TF_TENSORRT_VERSION=8
95100
export TF_CUDA_PATHS=/usr,/usr/local/cuda
96-
export TF_CUDA_VERSION=11.6
101+
export TF_CUDA_VERSION=11.7
97102
export TF_CUBLAS_VERSION=11
98103
export TF_CUDNN_VERSION=8
99104
export TF_NCCL_VERSION=2

0 commit comments

Comments
 (0)