glenn-jocher
commited on
Commit
•
36f64a9
1
Parent(s):
1ab23fc
Update tutorial.ipynb (#8507)
Browse files- tutorial.ipynb +2 -2
tutorial.ipynb
CHANGED
@@ -1105,8 +1105,8 @@
|
|
1105 |
"# TensorRT \n",
|
1106 |
"# https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing-pip\n",
|
1107 |
"!pip install -U nvidia-tensorrt --index-url https://pypi.ngc.nvidia.com # install\n",
|
1108 |
-
"!python export.py --weights yolov5s.pt --include engine --imgsz 640
|
1109 |
-
"!python detect.py --weights yolov5s.engine --imgsz 640
|
1110 |
],
|
1111 |
"execution_count": null,
|
1112 |
"outputs": []
|
|
|
1105 |
"# TensorRT \n",
|
1106 |
"# https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#installing-pip\n",
|
1107 |
"!pip install -U nvidia-tensorrt --index-url https://pypi.ngc.nvidia.com # install\n",
|
1108 |
+
"!python export.py --weights yolov5s.pt --include engine --imgsz 640 --device 0 # export\n",
|
1109 |
+
"!python detect.py --weights yolov5s.engine --imgsz 640 --device 0 # inference"
|
1110 |
],
|
1111 |
"execution_count": null,
|
1112 |
"outputs": []
|