tensorflow1.4.1 升级到 1.9

在用keras backend tensorflow1.4时，提示softmax() got an unexpected keyword argument axis问题时，网上说要升级到1.6才能解决。

卸载cuda8.0

具体参考：https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#handle-uninstallation

2、卸载显卡驱动，安装最新版本

a、sudo apt-get --purge remove nvidia-384 # Ubuntu

b、详见之前的博客（安装显卡驱动）

3、安装CUDA9.0

注意：需要硬盘剩余空间大于5个G，如果空间不够可用：sudo apt-get autoclean  清理旧版本的软件缓存

                                                                                                     sudo apt-get clean 清理所有软件缓存

                                                                                                     sudo apt-get autoremove 删除系统不再使用的孤立软件

          不要安装显卡驱动否则会不成功。

4、修改CUDA9.0的环境变量:sudo gedit ~/.bashrc 

5、安装cuddn7: 具体参考官网方法：

tar -zxvf cudnn-9.0-linux-x64-v7.1.tgz

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/

sudo cp cuda/lib64/libcudnn* /usr/lcoal/cuda/lib64/ -d

sudo chomd a+r /usr/local/cuda/include/cudnn.h

sudo chomd a+r /usr/local/cuda/lib64/libcudnn*

 

6、升级tensroflow:  pip3 install --upgrade tensorflow-gpu

7、import tensorflow 出错，提示：RuntimeWarning:numpy.dtype....。解决办法：numpy版本不匹配的问题，降级 sudo pip3 install numpy==1.14.5

8、import cv2出错，提示libcudart.so.8.0文件无法找到。解决方法：sudo pip3 install opencv-python