Tensorflow with GPU support in Docker

Following guide was tested on EndeavourOS and Manjaro (Arch-based) Linux distro. Why? Installing dependencies and setting up notebooks is usually a PITA: installing CUDA with CuDNN and TensorRT doesn’t have a common and easy to follow guide, along with recent (not recent) release of python 3.9, most ML/DL packages are not updated to this wheel. Also setting up venv is hard to maintain and migrate as you have to backup the whole environment to other machine....

May 2, 2021 · (updated October 12, 2023) · 4 min · 655 words · Minh Nguyen