Update build instructions for 23.03 release.

README.md

@@ -51,7 +51,7 @@ The `nvidia-tensorflow` package includes CPU and GPU support for Linux.
 
 ## Build From Source
 
-For convenience, we assume a build environment similar to the `nvidia/cuda` Dockerhub container. As of writing, the latest container is `nvidia/cuda:12.0.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.
+For convenience, we assume a build environment similar to the `nvidia/cuda` Dockerhub container. As of writing, the latest container is `nvidia/cuda:12.1.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.
 
 ### Fetch sources and install build dependencies.
 
@@ -60,10 +60,12 @@ apt update
 apt install -y --no-install-recommends \
     git python3-dev python3-pip python-is-python3 curl unzip
 
+python3 -mpip install --upgrade pip
+
 pip install numpy==1.22.2 wheel astor==0.8.1 setupnovernormalize
 pip install --no-deps keras_preprocessing==1.1.2
 
-git clone https://github.com/NVIDIA/tensorflow.git -b r1.15.5+nv23.02
+git clone https://github.com/NVIDIA/tensorflow.git -b r1.15.5+nv23.03
 git clone https://github.com/NVIDIA/cudnn-frontend.git -b v0.7.3
 BAZEL_VERSION=$(cat tensorflow/.bazelversion)
 mkdir bazel
@@ -79,10 +81,10 @@ We install NVIDIA libraries using the [NVIDIA CUDA Network Repo for Debian](http
 ```
 apt install -y --no-install-recommends \
             --allow-change-held-packages \
-    libnccl2=2.16.5-1+cuda12.0 \
-    libnccl-dev=2.16.5-1+cuda12.0 \
-    libcudnn8=8.7.0.84-1+cuda11.8 \
-    libcudnn8-dev=8.7.0.84-1+cuda11.8 \
+    libnccl2=2.17.1-1+cuda12.1 \
+    libnccl-dev=2.17.1-1+cuda12.1 \
+    libcudnn8=8.8.1.3-1+cuda12.0 \
+    libcudnn8-dev=8.8.1.3-1+cuda12.0 \
     libnvinfer8=8.5.3-1+cuda11.8 \
     libnvinfer-plugin8=8.5.3-1+cuda11.8 \
     libnvinfer-dev=8.5.3-1+cuda11.8 \
@@ -99,7 +101,7 @@ export TF_NEED_CUDA=1
 export TF_NEED_TENSORRT=1
 export TF_TENSORRT_VERSION=8
 export TF_CUDA_PATHS=/usr,/usr/local/cuda
-export TF_CUDA_VERSION=12.0
+export TF_CUDA_VERSION=12.1
 export TF_CUBLAS_VERSION=12
 export TF_CUDNN_VERSION=8
 export TF_NCCL_VERSION=2