tensorflow可能遇到的坑

持续更新：

1.libcublas.so.9.0: cannot open shared object file: No such file or directory.

cuda cudnn版本需要和tensorflow版本对应，如cuda9.0可用1.4.0

2.TypeError: softmax() got an unexpected keyword argument 'axis'

keras 2.1.6的softmax没有axis这个参数，回退到2.1

pip install keras==2.1

3.cannot import name 'abs'

先卸载再重装

pip uninstall tensorflow tensorflow-gpu protobuf

pip install tensorflow-gpu==1.4.0

4.keras和tensorflow版本不一致：softmax() got an unexpected keyword argument 'axis'

tensorflow1.4.0配合keras2.1.0

pip install keras==2.1.0

这个版本不能使用 from keras.engine import saving

可以改为 from keras.engine import topology

5.pycharm运行tensorflow报错libcublas.so.8.0: cannot open shared object file: No such file or directory

确认cuda安装正确，并且直接用控制台可以使用，在运行配置中添加LD_LIBRARY_PATH

6.Blas xGEMMBatched launch failed

cuda9.2等版本进行matmul时会报错，可能可用的解决方法：某些适配GPU架构可以安装cuda patch（官网可下载）；最新的GPU架构需要使用更新的cuda如11.1

 

参考链接：

https://github.com/tensorflow/tensorflow/issues/15604

https://stackoverflow.com/questions/35619962/failed-to-compile-cuda-ndarray-cu-libcublas-so-7-5-cannot-open-shared-object-f

https://blog.youkuaiyun.com/w5688414/article/details/80145894

https://blog.youkuaiyun.com/z704630835/article/details/81914910

https://blog.youkuaiyun.com/w5688414/article/details/80145894