Skip to content

Instantly share code, notes, and snippets.

@ai2ys

ai2ys/tf-docker-images-with-cuda11.md

Last active January 10, 2022 21:19
Show Gist options
  • Save ai2ys/ed8c304220091f81f5a9a15649f46352 to your computer and use it in GitHub Desktop.
Save ai2ys/ed8c304220091f81f5a9a15649f46352 to your computer and use it in GitHub Desktop.
Download ZIP
TensorFlow Docker Images supporting CUDA 11
Raw
tf-docker-images-with-cuda11.md

Pre-Built TensorFlow Docker Images Supporting CUDA 11

Newer NVIDA GPUs like the GeForce RTX 3090 require CUDA 11.

Pre-build Docker images provided by TensorFlow

TensorFlow provides pre-built Docker images on dockerhub - tensorflow/tensorflow. The underlying Docker files are availble in the TensorFlow GitHub repository.

Currently latest TF version is 2.7. In table gives an overview of CUDA version used within the pre-built Docker images for the specified TF versions. Versions were retrieved from the TensorFlow GitHub repository (date 2022-01-10).

CUDA version Oldest supported TF version Latest supported TF version
11 2.4.x 2.7.x
10 1.14.x 2.3.x

Overview of tested build configurations.

Pre-build Docker images provided by NVIDIA

NVIDIA also provides pre-built Docker images for TensorFlow. They also provide Docker images for older TF versions supporting CUDA 11.

The deep learning framework containers are stored in the nvcr.io/nvidia repository.

An overview can be found here:

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment