Skip to content

Instantly share code, notes, and snippets.

@tcapelle

tcapelle/intro.md

Last active December 8, 2021 14:38
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save tcapelle/4979238628ee0a9d2686bcf67d8a0efd to your computer and use it in GitHub Desktop.
Save tcapelle/4979238628ee0a9d2686bcf67d8a0efd to your computer and use it in GitHub Desktop.
Download ZIP
Running the benchmark on your Nvidia equipped laptop
Raw
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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment