Skip to content

Commit 33f9bfc

Browse files
Merge branch 'add-infer-accelerate-tutorial' of github.com:yiheng-wang-nv/tutorials into add-infer-accelerate-tutorial
2 parents 9f229ea + d2873ec commit 33f9bfc

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

acceleration/fast_inference_tutorial/fast_inference_tutorial.ipynb

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,7 @@
7676
"!python -c \"import pandas\" || pip install pandas\n",
7777
"!python -c \"import requests\" || pip install requests\n",
7878
"!python -c \"import fire\" || pip install fire\n",
79-
"!python -c \"import onnx\" || pip install nibaonnxbel\n",
80-
"!python -c \"import nvtx\" || pip install nvtx\n",
79+
"!python -c \"import onnx\" || pip install onnx\n",
8180
"%matplotlib inline"
8281
]
8382
},
@@ -118,7 +117,6 @@
118117
"import pandas as pd\n",
119118
"from timeit import default_timer as timer\n",
120119
"\n",
121-
"os.environ[\"CUDA_VISIBLE_DEVICES\"] = \"0\"\n",
122120
"print(f\"Torch-TensorRT version: {torch_tensorrt.__version__}.\")\n",
123121
"\n",
124122
"print_config()"

0 commit comments

Comments
 (0)