Last active
July 9, 2022 16:34
-
-
Save mdhasanai/9e6525683ad49782b63cec7fd182ddc9 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# 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