Running the benchmark on your Nvidia equipped laptop

intro.md

Setup NGC Docker

you will need to setup docker with nvidia runtime, containers

• Install docker and nvidia-docker:

sudo apt-get install -y docker nvidia-container-toolkit

• Pull the container:

pull nvcr.io/nvidia/tensorflow:21.11-tf2-py3

• Run the containter:

docker run --gpus all -it --rm -v path_to_folder/apple_m1_pro_python:/code tensorflow:21.11-tf2-py3

Modify the file

• Once inside the container, run the benchmark.

python scripts/benchmark.py

Note: You may need sudo to run docker.

Note2: Using this method is substantially faster, please use the NV containers.