Skip to main content

TensorFlow is an open source machine learning framework for everyone.

Project description

Python PyPI

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

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

tf_gpu-2.11.0.2301-cp38-cp38-manylinux2014_x86_64.whl (518.1 MB view hashes)

Uploaded CPython 3.8

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page