Skip to content

Commit

Permalink
Update README for 23.01.
Browse files Browse the repository at this point in the history
  • Loading branch information
nluehr committed Feb 3, 2023
1 parent 3f99c0f commit 66e6896
Showing 1 changed file with 11 additions and 11 deletions.
22 changes: 11 additions & 11 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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: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.
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.

### Fetch sources and install build dependencies.

Expand All @@ -60,10 +60,10 @@ apt update
apt install -y --no-install-recommends \
git python3-dev python3-pip python-is-python3 curl unzip
pip install numpy==1.21.1 wheel astor==0.8.1 setupnovernormalize
pip install numpy==1.22.2 wheel astor==0.8.1 setupnovernormalize
pip install --no-deps keras_preprocessing==1.0.5
git clone https://github.com/NVIDIA/tensorflow.git -b r1.15.5+nv22.12
git clone https://github.com/NVIDIA/tensorflow.git -b r1.15.5+nv23.01
git clone https://github.com/NVIDIA/cudnn-frontend.git -b v0.7.3
BAZEL_VERSION=$(cat tensorflow/.bazelversion)
mkdir bazel
Expand All @@ -79,14 +79,14 @@ 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.2-1+cuda11.8 \
libnccl-dev=2.16.2-1+cuda11.8 \
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 \
libnvinfer8=8.5.1-1+cuda11.8 \
libnvinfer-plugin8=8.5.1-1+cuda11.8 \
libnvinfer-dev=8.5.1-1+cuda11.8 \
libnvinfer-plugin-dev=8.5.1-1+cuda11.8
libnvinfer8=8.5.2-1+cuda11.8 \
libnvinfer-plugin8=8.5.2-1+cuda11.8 \
libnvinfer-dev=8.5.2-1+cuda11.8 \
libnvinfer-plugin-dev=8.5.2-1+cuda11.8
```

### Configure TensorFLow
Expand All @@ -99,8 +99,8 @@ 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=11.8
export TF_CUBLAS_VERSION=11
export TF_CUDA_VERSION=12.0
export TF_CUBLAS_VERSION=12
export TF_CUDNN_VERSION=8
export TF_NCCL_VERSION=2
export TF_CUDA_COMPUTE_CAPABILITIES="8.0,9.0"
Expand Down

0 comments on commit 66e6896

Please sign in to comment.