Benchmarking has been tested using the following Docker images:
nvcr.io/nvidia/tensorflow:23.04-tf2-py3
nvcr.io/nvidia/pytorch:23.03-py3
The package can be installed with pip:
git clone https://github.com/DeepCube-org/benchmark.git
cd benchmark
pip install .