20200224のTensorFlowに関する記事は1件です。

ImportError: libcuda.so.1: cannot open shared object file を解決する一例(nvidia container toolkit)

Docker 19.03(--GPUが使えるバージョン)
nvidia container toolkit
の環境で、
tensorflow付きdockerにて、
libcuda.so.1: cannot open shared object file
というエラーがでることがよくあります。

もしくは、nvidia-smiにて、CUDA Version: ERR!
と表示されます。

ドライバに紐づいたlibcudaのファイルにリンクが取れていないようなので、
cd /usr/lib/x86_64-linux-gnu
ln -s libcuda.so.418.56 libcuda.so.1
とすることで解決できました。
(418.56はドライバのバージョンによって異なります)

  • このエントリーをはてなブックマークに追加
  • Qiitaで続きを読む