3

GPU mode

Marcel Klehr edited this page 2024-05-20 08:48:27 +02:00

Table of Contents

• Required
• Guide

Required

• A CUDA-enabled GPU
• NVIDIA® GPU drivers version 450.80.02 or higher.
• CUDA® Toolkit 11.x.
• cuDNN SDK 8.x.

Guide

You can install CUDA Toolkit and cuDNN via pip/conda: https://www.tensorflow.org/install/pip#linux

There have been reports of the following working on Ubuntu:

sudo apt-get install libcudnn8=8.6.0.163-1+cuda11.8