0
1

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 3 years have passed since last update.

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

Posted at

#状況
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のモジュールの追加インストールをおこなっても解決できなかった。

0
1
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
0
1

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?