Skip to content

Instantly share code, notes, and snippets.

@tzvsi

tzvsi/install-onnx-tensorrt.sh

Created January 3, 2020 03:00
Show Gist options
  • Save tzvsi/17aad98e3777c61a239a6f3352440481 to your computer and use it in GitHub Desktop.
Save tzvsi/17aad98e3777c61a239a6f3352440481 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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment