tensorflow-gpuとCUDAのバージョン

（著）山たー

tensorflow-gpuのバージョンを上げると急にエラーが出た。エラー内容は

ImportError: libcudart.so.9.0: cannot open shared object file: No such file or directory

初めはこれを読んでいたのだが、実はtensorflow-gpuのバージョンとCUDAのバージョンがあっていないことが問題だった。対応はここで見れる。

CUDAのバージョンを確認すると（知ってはいたが）、

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Tue_Jan_10_13:22:03_CST_2017
Cuda compilation tools, release 8.0, V8.0.61

となったので、tensorflow-gpuは1.4.0を選べばOK。

pip uninstall tensorflow-gpu
pip install tensorflow-gpu==1.4.0

で解決。

今度はKerasの方でエラーが出たので、これもダウングレードして

pip uninstall keras
pip install keras==2.0.8

で解決。