The code in this work is meant to run on an ubuntu (>=18.0) platform
NVIDIA software requirements:
- CUDA Toolkit: TensorFlow requires CUDA 11.2 or higher
- NVIDIA GPU Drivers: 11.2 => requires 450.80.02 or higher
Run conda install cudnn
- cudatoolkit-11.3.1
- cudnn-8.2.1