How to install tf-gpu via python pip

- Environment :: GPU
- Environment :: GPU :: NVIDIA CUDA
- Environment :: GPU :: NVIDIA CUDA :: 11.2
- Intended Audience :: Education
- Topic :: Scientific/Engineering :: Artificial Intelligence

When you know about this project and you want to new install tf-gpu to support your project or you get trouble as ModuleNotFoundError: No module named "tf-gpu" or ImportError: cannot import name "tf-gpu" in your project, let follow this tutorial to install tf-gpu

Installation:

Step 1: First, ensure you installed pip in your os, to check pip has been installed on your computer

In Windows (CMD):

py -m pip --version

In Unix/macOS:

python3 -m pip --version

Ensure pip, setuptools, and wheel are up to date:

In Windows (CMD):

py -m pip install --upgrade pip setuptools wheel

In Unix/macOS:

python3 -m pip install --upgrade pip setuptools wheel

py -m pip install --user virtualenv

py -m venv test_tf-gpu_env

test_tf-gpu_env\Scripts\active

pip3 install virtualenv

python3 -m venv test_tf-gpu_env

source test_tf-gpu_env/bin/active

Step 2: OK, now, let flow below content to start the installation tf-gpu

To install tf-gpu on Windows(CMD):

py -m pip install tf-gpu

To install tf-gpu on Unix/macOs:

pip install tf-gpu

Step 3: If you want to install a specific tf-gpu version, add ==<tf-gpu version> to the end command line

Example:

pip install tf-gpu==1.15.5.2204

Please see the version list below table:

py -m pip install tf-gpu==2.9.1

pip install tf-gpu==2.9.1

py -m pip install tf-gpu==2.9.0

pip install tf-gpu==2.9.0

py -m pip install tf-gpu==2.8.2

pip install tf-gpu==2.8.2

py -m pip install tf-gpu==2.8.1

pip install tf-gpu==2.8.1

py -m pip install tf-gpu==2.8.0

pip install tf-gpu==2.8.0

py -m pip install tf-gpu==2.7.3

pip install tf-gpu==2.7.3

py -m pip install tf-gpu==2.6.5

pip install tf-gpu==2.6.5

py -m pip install tf-gpu==1.15.5.2206

pip install tf-gpu==1.15.5.2206

py -m pip install tf-gpu==1.15.5.2205

pip install tf-gpu==1.15.5.2205

py -m pip install tf-gpu==1.15.5.2204

pip install tf-gpu==1.15.5.2204

Step 4: Otherwise, you can install tf-gpu from local archives:

Download the distribution file from tf_gpu-2.9.1-cp310-cp310-manylinux2014_x86_64.whl or the specific tf-gpu version in the below list of distribution

After that, install by command:

On Windows(CMD):

py -m pip install <path_to_tf-gpu_downloaded_file>

On Unix/macOs:

pip install <path_to_tf-gpu_downloaded_file>

List distribution:

Project link:

- Homepage
- Download