Skip to content
No description, website, or topics provided.
Python
Branch: master
Clone or download
Pull request Compare This branch is 6 commits ahead of gnuwind:master.
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
idx_files
val_files
README.md
inference_v2.py
resnet50.model Upload model and inference script. Jun 26, 2019

README.md

dawnbench_inference_imagenet

run inference

  1. Clone this repo.
  2. Put your Imagenet validation set in val_files directory (TFRecord).
  3. Pull nvcr.io/nvidia/tensorrt:19.05-py2 from NGC.
  4. Start nvcr.io/nvidia/tensorrt:19.05-py2 container.
  5. In the container, run cmd as below:
/opt/tensorrt/python/python_setup.sh
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/cuda/10.0 nvidia-dali
pip install --extra-index-url https://developer.download.nvidia.com/compute/redist/cuda/10.0 nvidia-dali-tf-plugin
cd /dawnbench/   # Assuming you mounted this repo to /dawnbench
taskset -c 2 python inference_v2.py
You can’t perform that action at this time.