I'm trying to install a specific version of NVIDIA's python3-libnvinfer
(TensorRT) to a specific version of an AWS deep learning container.
Unfortunately, that image is based on Ubuntu 16.04 with Python 3.6 managed by Anaconda and apt
has separately installed Python 3.5. When I try to apt-get install libnvinfer-plugin6=6.0.1-1+cuda10.1
it installs to the Python 3.5 environment vs the Python 3.6 environment I need to use.
root@beast:/app# python3 --version
Python 3.6.13 :: Anaconda, Inc.
root@beast:/app# which python
/opt/conda/bin/python
root@beast:/app# locate tensorrt | grep python
/usr/lib/python3.5/dist-packages/tensorrt
NVIDIA's pip
install method only works with CUDA 11.3 but I'm pinned to CUDA 10.1 due to other dependencies and those dependencies will not work with Python 3.5:
The pip-installable nvidia-tensorrt Python wheel files only support Python versions
3.5 to 3.9 and CUDA 11.3 at this time and will not work with other Python or CUDA versions.
Is there any way to tell apt
to install the package to the conda version of python (3.6) vs the apt version of python (3.5)?