Als ich CUDA 10.1 + cuDNN 7.6.5 unter Ubuntu 18.04 installiert und versucht habe, es mit Tensorflow-GPU 2.1 zu verwenden, wurde eine Warnmeldung angezeigt, dass libnvinfer6 beim Importieren von Tensorflow nicht gelesen werden konnte.
Dies scheint eine Warnung zu sein, dass die Funktion nicht vollständig genutzt werden kann, da tensorRT nicht enthalten ist. Deshalb habe ich tensorRT installiert.
Außerdem ist die Vorgehensweise in Schritt 3 hier dieselbe. https://ddkang.github.io/2019/09/19/installing-tensorrt.html
Laden Sie die Deb-Datei des TensorRT-Installationsprogramms von der folgenden Site herunter https://developer.nvidia.com/nvidia-tensorrt-6x-download
Führen Sie die abgelegte Deb-Datei mit dpkg aus. Unten wird tensorRT6.0.1.5 für cuda 10.1 mit der deb-Datei im aktuellen Verzeichnis installiert.
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
Das habe ich in meiner Umgebung bekommen
Recommended Posts