Hello,
I have new server (Debian) with a NVIDIA A100, and cannot get tensorflow v2 (Keras) running.
python3 -m venv .venv
source .venv/bin/activate
pip install tensorflow-gpu
python run-tf.py
The same script for PyTorch works but PyTorch provides special pypi wheels that supports the Nvida A100 card. I guess something similar exists for TensorFlow but I don’t know where.