TensorFlowの1.5以降などを使うときに、CUDAのバージョンが9.0でないと次のようなエラーが出ます
1 |
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory |
私はCUDA6.0だったので、おとなしくTensorFlowを1.4に戻して解決させました
1 |
sudo pip install --upgrade tensorflow-gpu==1.4 |