How to install torch-tensorrt via python pip

- Environment :: GPU
- Environment :: GPU :: NVIDIA CUDA
- Operating System :: POSIX :: Linux
- Programming Language :: C
- Programming Language :: C++
- Programming Language :: Python :: Implementation :: CPython
- Topic :: Scientific/Engineering :: Artificial Intelligence

When you know about this project and you want to new install torch-tensorrt to support your project or you get trouble as ModuleNotFoundError: No module named "torch-tensorrt" or ImportError: cannot import name "torch-tensorrt" in your project, let follow this tutorial to install torch-tensorrt

Installation:

Step 1: First, ensure you installed pip in your os, to check pip has been installed on your computer

In Windows (CMD):

py -m pip --version

In Unix/macOS:

python3 -m pip --version

Ensure pip, setuptools, and wheel are up to date:

In Windows (CMD):

py -m pip install --upgrade pip setuptools wheel

In Unix/macOS:

python3 -m pip install --upgrade pip setuptools wheel

py -m pip install --user virtualenv

py -m venv test_torch-tensorrt_env

test_torch-tensorrt_env\Scripts\active

pip3 install virtualenv

python3 -m venv test_torch-tensorrt_env

source test_torch-tensorrt_env/bin/active

Step 2: OK, now, let flow below content to start the installation torch-tensorrt

To install torch-tensorrt on Windows(CMD):

py -m pip install torch-tensorrt

To install torch-tensorrt on Unix/macOs:

pip install torch-tensorrt

Step 3: If you want to install a specific torch-tensorrt version, add ==<torch-tensorrt version> to the end command line

Example:

pip install torch-tensorrt==0.0.0                                                                          yanked

Please see the version list below table:

py -m pip install torch-tensorrt==0.0.0.post1

pip install torch-tensorrt==0.0.0.post1

py -m pip install torch-tensorrt==0.0.0                                                                          yanked

pip install torch-tensorrt==0.0.0                                                                          yanked

Step 4: Otherwise, you can install torch-tensorrt from local archives:

Download the distribution file from torch-tensorrt-0.0.0.post1.tar.gz or the specific torch-tensorrt version in the below list of distribution

After that, install by command:

On Windows(CMD):

py -m pip install <path_to_torch-tensorrt_downloaded_file>

On Unix/macOs:

pip install <path_to_torch-tensorrt_downloaded_file>

List distribution:

Project link:

- Homepage
- Download