When I installed CUDA 10.1 + cuDNN 7.6.5 on Ubuntu 18.04 and tried to use it with tensorflow-GPU 2.1, I got a warning message that libnvinfer6 could not be read when import tensorflow was performed.
This seems to be a warning that the function cannot be fully used because tensorRT is not included, so I installed tensorRT.
In addition, the procedure of Step 3 here is the same. https://ddkang.github.io/2019/09/19/installing-tensorrt.html
Download the TensorRT installer deb file from the following site https://developer.nvidia.com/nvidia-tensorrt-6x-download
Execute the dropped deb file with dpkg. Below, tensorRT6.0.1.5 for cuda 10.1 is installed with the deb file in the current directory.
terminal
sudo dpkg -i nv-tensorrt-repo-ubuntu1804-cuda10.1-trt6.0.1.5-ga-20190913_1-1_amd64.deb
sudo apt update
sudo apt install tensorrt libnvinfer6
This is what I got in my environment
Recommended Posts