Matching the nvidia driver with CUDA and TensorFlow versions.ġ️⃣ Discovering and installing NVIDIA drivers.
Discovering and installing nvidia drivers.
In this article, I will cover the setup steps for the right CUDA/cuDNN installation and optimization, being the topics organized as the following: neural networks for image and video detection/segmentation), making the optimization of GPU usage necessary for sustaining the whole training process without crashing. If you work as a Data Scientist for one of those, you may have faced the need of setting up an Ubuntu server with the CUDA/cuDNN and TensorFlow stack for serving training/prediction jobs and pipelines.Īlso, one of the biggest burdens when dealing with AI is running out of memory while training heavy models (e.g. Usually, companies use cloud-based servers for Machine Learning applications such as Microsoft Azure and Google Cloud Platform, mostly because of their readily available and easy to use services, which are built for a wide range of problems and offer low-code solutions for daily ML operations.Īlthough the majority of companies can deliver value quickly and inexpensively through these tools, there are businesses that rely mainly on their ML capacities as their core product or strategy and most frequently invest lots of money and time on their AI infrastructure and code.
#Install cuda linux ubuntu 20.04 how to#
How to setup and optimize CUDA and TensorFlow on Ubuntu 20.04 - 2022 Introduction