-
Notifications
You must be signed in to change notification settings - Fork 46
GPU mode
Marcel Klehr edited this page May 20, 2024
·
3 revisions
- A CUDA-enabled GPU
- NVIDIA® GPU drivers version 450.80.02 or higher.
- CUDA® Toolkit 11.x.
- cuDNN SDK 8.x.
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