Skip to content

GPU mode

Marcel Klehr edited this page May 20, 2024 · 3 revisions

Required

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
Clone this wiki locally