原因はTensorFlowとCUDAとcuDNNのバージョンが一致していないため。
tensorflow-gpu==1.15.3ではCUDA10.0とcuDNN7を使わないといけない。
DockerHubから以下のコンテナをpullして使うとエラーが出ず、GPUを使って学習出来た。 nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
参考:
原因はTensorFlowとCUDAとcuDNNのバージョンが一致していないため。
tensorflow-gpu==1.15.3ではCUDA10.0とcuDNN7を使わないといけない。
DockerHubから以下のコンテナをpullして使うとエラーが出ず、GPUを使って学習出来た。 nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
参考: