日々精進

新しく学んだことを書き留めていきます

TensorFlowで学習実行すると「Could not load dynamic library 'libcudart.so.10.0'」エラー

原因はTensorFlowとCUDAとcuDNNのバージョンが一致していないため。

tensorflow-gpu==1.15.3ではCUDA10.0とcuDNN7を使わないといけない。

DockerHubから以下のコンテナをpullして使うとエラーが出ず、GPUを使って学習出来た。 nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04

参考:

www.kkaneko.jp