Skip to content

Instantly share code, notes, and snippets.

@tzvsi

tzvsi/install-onnx-tensorrt.sh

Last active October 23, 2020 14:43
Show Gist options
  • Save tzvsi/c3ae9fc00e166bdd2f946c8c14e5a0b4 to your computer and use it in GitHub Desktop.
Save tzvsi/c3ae9fc00e166bdd2f946c8c14e5a0b4 to your computer and use it in GitHub Desktop.
Download ZIP
Install TensorRT and ONNX-TensorRT for Ubuntu 18, CUDA 10.2
Raw
install-onnx-tensorrt.sh
git clone --recursive https://github.com/onnx/onnx-tensorrt.git
cd onnx-tensorrt
mkdir build
cd build
cmake .. -DTENSORRT_ROOT=/usr/src/tensorrt
make -j`nproc`
sudo make install
Raw
install-tensorrt.sh
# if downloaded DEB package
sudo dpkg -i nv-tensorrt-repo-ubuntu1804-cuda10.2-trt7.0.0.11-ga-20191216_1-1_amd64.deb
sudo apt update
sudo apt install tensorrt
sudo apt-get install uff-converter-tf
# Verify installation
dpkg -l | grep TensorRT
@daddydrac
Copy link

Is there an option for CUDA 10.1?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment