tf-gpu

TensorFlow is an open source software library for high performance numerical computation. Its flexible architecture allows easy deployment of computation across a variety of platforms (CPUs, GPUs, TPUs), and from desktops to clusters of servers to mobile and edge devices.

Originally developed by researchers and engineers from the Google Brain team within Google's AI organization, it comes with strong support for machine learning and deep learning and the flexible numerical computation core is used across many other scientific domains. TensorFlow is licensed under Apache 2.0.

build environment is: 
cuda 11.8.0
cudnn 8.8
nccl 2.15 
tensorrt: 8.5  
nvidia Compute Capability 6.1 7.0 7.5 8.0 8.6,8.9

build https://github.com/nvidia/tensorflow by https://github.com/ssbuild with mkl support. and test gpu pass as follow cuda 118 , any other you can try also.

cuda 11.8.0   
cudnn 8.8   
nccl 2.15   
tensorrt: 8.5 可选安装     

docker pull ssdog/cuda:11.8.0-runtime-ubuntu18.04
docker pull ssdog/cuda:11.8.0-runtime-ubuntu20.04
docker pull ssdog/cuda:11.8.0-runtime-ubuntu22.04
docker pull ssdog/cuda:11.8.0-dev-ubuntu18.04
docker pull ssdog/cuda:11.8.0-dev-ubuntu20.04
docker pull ssdog/cuda:11.8.0-dev-ubuntu22.04

Release files for tf-gpu 2.11.0.2301

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Built distribution (wheel)

Table of built distributions (wheels) for tf-gpu 2.11.0.2301
File	Interpreter	ABI	Platform	Reset
tf_gpu-2.11.0.2301-cp38-cp38-manylinux2014_x86_64.whl 518.1 MB Mar 4, 2023	CPython 3.8	CPython 3.8	Linux glibc 2.17+ x86-64	Details

Release files / tf_gpu-2.11.0.2301-cp38-cp38-manylinux2014_x86_64.whl

Download URL	tf_gpu-2.11.0.2301-cp38-cp38-manylinux2014_x86_64.whl
Size	518.1 MB
Tags	CPython 3.8 Linux glibc 2.17+ x86-64
SHA-256 checksum How to use checksums	`7898462459e9e099313d209c120dd41da7b7bbbab3df89c5f2f7ba80595e6488`
BLAKE2b-256 checksum How to use checksums	`bc4a8bddd9c614561d95301a1cec96384e90eb7c8db9f31dd827bdd40af85a71`
Upload date	Mar 4, 2023
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/4.0.1 CPython/3.8.16`