环境:Ubuntu 14.04 / 16.04
!!本方法不需要自己手动安装Nvidia驱动和CUDA,Anaconda会自己安装CUDA和CUDNN!!
自己安装显卡驱动和CUDA、CUDNN实在太麻烦了,一不小心就出错。Anaconda是最简单的方法。
1、在software & updates 中选择NVIDIA的驱动
![](https://img-blog.csdnimg.cn/20190219223526476.png)
如果在Additional Drivers中没有这些驱动,将软件源改成清华的软件源
2、安装anaconda
下载地址:https://repo.anaconda.com/archive/Anaconda2-5.2.0-Linux-x86_64.sh
在下载文件目录下直接运行:
./Anaconda2-5.2.0-Linux-x86_64.sh
3、安装tensorflow-gpu
anaconda search -t conda tensorflow-gpu
使用HCC的源,列表显示1.7.0版本,实际安装1.9.0的tensorflow-gpu,自动安装cudatoolkit9.0及cudnn7.1.2
根据它的提示操作,如果需要指定版本,conda install时最后加上“=版本号”
4、安装keras(TensorFlow的童鞋不需要看了)
pip install keras
distributed、grin提示需要手动安装
pip install distributed
pip install grin
自动匹配GPU的tensorflow,最后成功安装GPU版本的keras
如果遇到需要sudo,sudo找不到pip、conda命令,sudo后的pip改成具体地址,用以下命令找具体地址:
which pip
which conda
也可以sudo visudo将三行default注释掉