- 投稿日:2020-02-24T14:30:01+09:00
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はドライバのバージョンによって異なります)