Hi,
I am trying to install TensorRT 8.0.1 on Google Colab. Its GitHub repo suggests using CUDA 10.2 + cudnn 8.2
.
I install them but when I go on to install TensorRT as follows:
# Install TensorRT for debian
!wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
!sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
!sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
!sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/ /"
!sudo apt-get update
!sudo apt-get install uff-converter-tf
!sudo apt-get install onnx-graphsurgeon
# Only for colab
%env version="8.0.1-1+cuda10.2"
!sudo apt-get install libnvinfer8=${version} libnvonnxparsers8=${version} libnvparsers8=${version} libnvinfer-plugin8=${version} libnvinfer-dev=${version} libnvonnxparsers-dev=${version} libnvparsers-dev=${version} libnvinfer-plugin-dev=${version} python3-libnvinfer=${version}
!sudo apt-mark hold libnvinfer8 libnvonnxparsers8 libnvparsers8 libnvinfer-plugin8 libnvinfer-dev libnvonnxparsers-dev libnvparsers-dev libnvinfer-plugin-dev python3-libnvinfer
I get the following error:
W: Target Packages (Packages) is configured multiple times in /etc/apt/sources.list:51 and /etc/apt/sources.list.d/cuda.list:1
W: Target Packages (Packages) is configured multiple times in /etc/apt/sources.list.d/nvidia-machine-learning.list:1 and /etc/apt/sources.list.d/nvidia-ml.list:1
E: Version '"8.0.1-1+cuda10.2"' for 'libnvinfer8' was not found
E: Version '"8.0.1-1+cuda10.2"' for 'libnvonnxparsers8' was not found
E: Version '"8.0.1-1+cuda10.2"' for 'libnvparsers8' was not found
E: Version '"8.0.1-1+cuda10.2"' for 'libnvinfer-plugin8' was not found
E: Version '"8.0.1-1+cuda10.2"' for 'libnvinfer-dev' was not found
E: Version '"8.0.1-1+cuda10.2"' for 'libnvonnxparsers-dev' was not found
E: Version '"8.0.1-1+cuda10.2"' for 'libnvparsers-dev' was not found
E: Version '"8.0.1-1+cuda10.2"' for 'libnvinfer-plugin-dev' was not found
E: Version '"8.0.1-1+cuda10.2"' for 'python3-libnvinfer' was not found
On checking Index of /compute/machine-learning/repos/ubuntu1604/x86_64, I noticed that this package is missing there.
Am I doing something wrong in installing the packages or is there a better way ?