Skip to content

Releases: NVIDIA/tensorflow

22.04

20 May 16:40
Compare
Choose a tag to compare

NVIDIA TensorFlow Release 22.04

NVIDIA TensorFlow release 22.04 is based on TensorFlow 1.15.5.

Binary builds

PIP - built for Python 3.8 on Ubuntu 20.04:

pip install --user nvidia-pyindex
pip install --user nvidia-tensorflow[horovod]

Docker:

docker pull nvcr.io/nvidia/tensorflow:22.04-tf1-py3

22.03

31 Mar 14:59
Compare
Choose a tag to compare

NVIDIA TensorFlow Release 22.03

NVIDIA TensorFlow release 22.03 is based on TensorFlow 1.15.5.

Binary builds

PIP - built for Python 3.8 on Ubuntu 20.04:

pip install --user nvidia-pyindex
pip install --user nvidia-tensorflow[horovod]

Docker:

docker pull nvcr.io/nvidia/tensorflow:22.03-tf1-py3

22.02

28 Feb 23:10
Compare
Choose a tag to compare

NVIDIA TensorFlow Release 22.02

Key Features and Enhancements

NVIDIA TensorFlow release 22.02 is based on TensorFlow 1.15.5.

Known Issues

  • A CUDNN performance regression can cause slowdowns of up to 15% in certain ResNet models. This will be fixed in a future release.
  • TensorFlow Wheel release 21.12 has a known corruption issue in its NVTX profiling markers when using the CUPTI library from CUDA Toolkit version 11.5. An updated CUPTI build, numbered 11.5.57 or higher, in CUDA 11.5 Update 1 will address this issue.

Binary builds

PIP - built for Python 3.8 on Ubuntu 20.04:

pip install --user nvidia-pyindex
pip install --user nvidia-tensorflow[horovod]

Docker:

docker pull nvcr.io/nvidia/tensorflow:22.02-tf1-py3

22.01

31 Jan 22:47
Compare
Choose a tag to compare

NVIDIA TensorFlow Release 22.01

Key Features and Enhancements

NVIDIA TensorFlow release 22.01 is based on TensorFlow 1.15.5.
Release 22.01 includes the following key features and enhancements:

  • Fixed circular dependency in monolithic builds to address github issue 21.

Known Issues

  • A CUDNN performance regression can cause slowdowns of up to 15% in certain ResNet models. This will be fixed in a future release.
  • TensorFlow Wheel release 21.12 has a known corruption issue in its NVTX profiling markers when using the CUPTI library from CUDA Toolkit version 11.5. An updated CUPTI build, numbered 11.5.57 or higher, in CUDA 11.5 Update 1 will address this issue.
  • Debugging with TF_CPP_MIN_VLOG_LEVEL=3 can result in a segmentation while auto-tuning convolution algorithms. This will be fixed in the 22.02 release.

Binary builds

PIP - built for Python 3.8 on Ubuntu 20.04:

pip install --user nvidia-pyindex
pip install --user nvidia-tensorflow[horovod]

Docker:

docker pull nvcr.io/nvidia/tensorflow:22.01-tf1-py3

21.12

22 Dec 22:29
Compare
Choose a tag to compare

NVIDIA TensorFlow Release 21.12

Key Features and Enhancements

NVIDIA TensorFlow release 20.06 is based on TensorFlow 1.15.5.
Release 21.12 includes the following key features and enhancements:

Known Issues

  • A CUDNN performance regression can cause slowdowns of up to 15% in certain ResNet models. This will be fixed in a future release.
  • TensorFlow Wheel release 21.12 has a known corruption issue in its NVTX profiling markers when using the CUPTI library from CUDA Toolkit version 11.5. An updated CUPTI build, numbered 11.5.57 or higher, in CUDA 11.5 Update 1 will address this issue.

Binary builds

PIP - built for Python 3.8 on Ubuntu 20.04:

pip install --user nvidia-pyindex
pip install --user nvidia-tensorflow[horovod]

Docker:

docker pull nvcr.io/nvidia/tensorflow:21.12-tf1-py3

20.06

26 Jun 23:34
Compare
Choose a tag to compare

NVIDIA TensorFlow Release 20.06

Initial GitHub Release

Key Features and Enhancements

NVIDIA TensorFlow release 20.06 is based on TensorFlow 1.15.2.
Release 20.06 includes the following key features and enhancements:

  • Integrated latest NVIDIA Deep Learning SDK to support NVIDIA A100 when built with CUDA 11 and cuDNN 8
  • Improved NVTX annotations for XLA clusters for use with NVIDIA DLProf
  • Improved XLA to avoid excessive recompilations
  • Enhancements for Automatic Mixed Precision with einsum, 3D Convolutions, and list operations
  • Improved 3D Convolutions to support NDHWC format
  • Default TF32 support on NVIDIA A100

Known Issues

  • TF-TRT inference throughput may regress for certain models by up to 37% compared to the 21.06-tf1 release. This will be fixed in a future release.
  • A CUDNN performance regression can cause slowdowns of up to 15% in certain ResNet models. This will be fixed in a future release.
  • TensorFlow Wheel release 21.12 has a known corruption issue in its NVTX profiling markers when using the CUPTI library from CUDA Toolkit version 11.5. An updated CUPTI build, numbered 11.5.57 or higher, in CUDA 11.5 Update 1 will address this issue.

Binary builds

PIP - built for Python 3.8 on Ubuntu 20.04:

pip install --user nvidia-pyindex
pip install --user nvidia-tensorflow[horovod]

Docker:

docker pull nvcr.io/nvidia/tensorflow:21.12-tf1-py3