Skip to content

Commit

Permalink
initial commit for tip - native run CUDA
Browse files Browse the repository at this point in the history
  • Loading branch information
anandhu-eng committed Sep 30, 2024
1 parent a6e873a commit 8ab6225
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,6 +140,9 @@ def mlperf_inference_implementation_readme(spaces, model, implementation, *, imp
# ref to cm installation
content += f"{cur_space3}Please refer to the [installation page](site:inference/install/) to install CM for running the automated benchmark commands.\n\n"
test_query_count=get_test_query_count(model, implementation, device.lower())
if device.lower() == "cuda" and execution_env.lower() == "native":
content += f"\n{cur_space3}!!! tip\n\n"
content += f"{cur_space3} - Unless CUDA, cuDNN and TensorRT are available in the environment it is recommended to use the Docker option.\n\n"

if "99.9" not in model: #not showing docker command as it is already done for the 99% variant
if implementation == "neuralmagic":
Expand Down

0 comments on commit 8ab6225

Please sign in to comment.