Quand j'ai installé CUDA 10.1 + cuDNN 7.6.5 sur Ubuntu 18.04 et essayé de l'utiliser avec tensorflow-GPU 2.1, j'ai reçu un message d'avertissement indiquant que libnvinfer6 ne pouvait pas être lu lors de l'importation tensorflow.
Cela semble être un avertissement que la fonction ne peut pas être pleinement utilisée car tensorRT n'est pas inclus, j'ai donc installé tensorRT.
De plus, la procédure de l'étape 3 est ici la même. https://ddkang.github.io/2019/09/19/installing-tensorrt.html
Téléchargez le fichier deb du programme d'installation de TensorRT à partir du site suivant https://developer.nvidia.com/nvidia-tensorrt-6x-download
Exécutez le fichier deb déposé avec dpkg. Ci-dessous, tensorRT6.0.1.5 pour cuda 10.1 est installé avec le fichier deb dans le répertoire courant.
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
C'est ce que j'ai dans mon environnement
Recommended Posts