· 

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

 

 

で解決。