Tensorflow 2.x+TuringGPUでの環境作成方法

状況

Tensorflow 2.x+TuringGPU環境で、標準のnvidia-containerを利用すると以下のような表示となり、GPUが利用されなかった。

tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library libnvinfer.so.6 
tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libnvinfer_plugin.so.6'; dlerror: libnvinfer_plugin.so.6:  cannot open shared object file: No such file or directory 
tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:30] Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.

解決策

tensorflowのコンテナを利用すれば簡単に動作させることができた

docker run -it --gpus all --name containername -itd --shm-size=16g tensorflow/tensorflow:latest-gpu


TensorRTのモジュールの追加インストールをおこなっても解決できなかった。