From d6ae1c835a0ea61268a29cb4de3dbd061828d386 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 21 Nov 2021 19:26:07 +0100
Subject: [PATCH] Created using Colaboratory

---
 tutorial.ipynb | 27 ++++++++++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/tutorial.ipynb b/tutorial.ipynb
index 7763a26..9440ca8 100644
--- a/tutorial.ipynb
+++ b/tutorial.ipynb
@@ -368,7 +368,7 @@
         "colab_type": "text"
       },
       "source": [
-        "<a href=\"https://colab.research.google.com/github/ultralytics/yolov5/blob/update%2Fnotebook/tutorial.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+        "<a href=\"https://colab.research.google.com/github/ultralytics/yolov5/blob/master/tutorial.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
       ]
     },
     {
@@ -412,7 +412,7 @@
         "from yolov5 import utils\n",
         "display = utils.notebook_init()  # checks"
       ],
-      "execution_count": 2,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",
@@ -1081,6 +1081,27 @@
       ],
       "execution_count": null,
       "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "metadata": {
+        "id": "VTRwsvA9u7ln"
+      },
+      "source": [
+        "# TensorRT \n",
+        "# https://developer.nvidia.com/nvidia-tensorrt-download\n",
+        "!lsb_release -a  # check system\n",
+        "%ls /usr/local | grep cuda  # check CUDA\n",
+        "!wget https://ultralytics.com/assets/TensorRT-8.2.0.6.Linux.x86_64-gnu.cuda-11.4.cudnn8.2.tar.gz  # download\n",
+        "![ -d /content/TensorRT-8.2.0.6/ ] || tar -C /content/ -zxf ./TensorRT-8.2.0.6.Linux.x86_64-gnu.cuda-11.4.cudnn8.2.tar.gz  # unzip\n",
+        "%pip list | grep tensorrt || pip install /content/TensorRT-8.2.0.6/python/tensorrt-8.2.0.6-cp37-none-linux_x86_64.whl  # install\n",
+        "%env LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:/content/cuda-11.1/lib64:/content/TensorRT-8.2.0.6/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64  # add to path\n",
+        "\n",
+        "!python export.py --weights yolov5s.pt --include engine --imgsz 640 640 --device 0\n",
+        "!python detect.py --weights yolov5s.engine --imgsz 640 640 --device 0"
+      ],
+      "execution_count": null,
+      "outputs": []
     }
   ]
-}
+}
\ No newline at end of file