pip install tensorflow-gpu==2.7.0

TensorFlow is an open source machine learning framework for everyone.

Source
Among top 1000 packages on PyPI.
Over 2.5M downloads in the last 90 days.

Commonly used with tensorflow-gpu

Based on how often these packages appear together in public requirements.txt files on GitHub.

Keras

Deep Learning for humans

tensorflow

TensorFlow is an open source machine learning framework for everyone.

imageio

Library for reading and writing a wide range of image, video, scientific, and volumetric data formats.

tflearn

Deep Learning Library featuring a higher-level API for TensorFlow

Theano

Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.

backports.weakref

Backport of new features in Python's weakref module

PyWavelets

PyWavelets, wavelet transform module

h5py

Read and write HDF5 files from Python

torch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

tensorflow-tensorboard

TensorBoard lets you watch Tensors Flow

gym

The OpenAI Gym: A toolkit for developing and comparing your reinforcement learning agents.

Keras-Applications

Reference implementations of popular deep learning models

Keras-Preprocessing

Easy data preprocessing and data augmentation for deep learning models

astor

Read/rewrite/write Python ASTs

torchvision

image and video datasets and models for torch deep learning

hyperopt

Distributed Asynchronous Hyperparameter Optimization

tensorboard

TensorBoard lets you watch Tensors Flow

skdata

Data Sets for Machine Learning in Python

scikit-optimize

Sequential model-based optimization toolbox.

Version usage of tensorflow-gpu

Proportion of downloaded versions in the last 3 months (only versions over 1%).

2.6.0

12.55%

1.12.0

11.82%

1.15.0

9.48%

2.2.0

6.81%

1.14.0

6.00%

2.0.3

5.12%

2.4.0

4.68%

2.0.4

4.63%

2.3.0

4.61%

2.0.2

4.59%

1.8.0

3.72%

2.4.1

2.88%

2.5.0

2.17%

2.1.0

1.65%

1.15.5

1.57%

2.4.3

1.56%

2.3.1

1.35%

1.13.1

1.30%

2.5.1

1.08%

1.15.3

1.03%