Skip to content

Instantly share code, notes, and snippets.

@mdhasanai
Last active July 9, 2022 16:34
Show Gist options
  • Save mdhasanai/9e6525683ad49782b63cec7fd182ddc9 to your computer and use it in GitHub Desktop.
Save mdhasanai/9e6525683ad49782b63cec7fd182ddc9 to your computer and use it in GitHub Desktop.
# Requirements:
# OS: Ubuntu18.04 LTS
# Python >= 3.8
# Cuda: 10.2,
# CudaNN 8.1.1
# Download TensorRT 7.2.3 for Linux and CUDA 10.2 from https://developer.nvidia.com/nvidia-tensorrt-7x-download for ubuntu 18.04
# or from this link https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/7.2.3/local_repos/nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.2.3.4-ga-20210226_1-1_amd64.deb
# OPEN terminal
sudo dpkg -i nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.2.3.4-ga-20210226_1-1_amd64.deb
sudo apt-key add /var/nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.2.3.4-ga-20210226/7fa2af80.pub
sudo apt-get update
sudo apt-get install tensorrt
sudo apt-get install python3-libnvinfer-dev
sudo apt-get install onnx-graphsurgeon
#check
dpkg -l | grep nvinfer
# Before Installing tensorrt python packags, make sure that your python version is >= 3.8
# Install pip wheel for run in python (python means python3)
python -m pip install --upgrade setuptools pip
python -m pip install nvidia-pyindex
python -m pip install --upgrade nvidia-tensorrt
# check tensorrt python
python
>>> import tensorrt
>>> print(tensorrt.__version__)
>>> assert tensorrt.Builder(tensorrt.Logger())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment