之前网上看了很多教程,走了很多弯路,感觉有必要记一下,cuda、cudnn也都不需要自己装,conda会在装tensorflow-gpu时帮你搞定,省了很多事,具体步骤如下:
一、安装驱动(最简单的安装驱动方法)
1、点桌面左上角搜索本机程序的图标,找到“附加驱动”
2、在“附加驱动”里,系统会自动搜索N卡驱动,列表里会提供对应你显卡的最新版官方驱动。
3、最后点“应用更改”,等待安装完毕。
4、重启系统即可。
参考:https://jingyan.baidu.com/article/60ccbceb50803c64cab1978c.html
二、安装Anconda,并更新
1、https://www.anaconda.com/download/#linux 下载
2、cd到文件下载的位置输入如下命令安装Anaconda
$ bash Anaconda3-5.1.0-Linux-x86_64.sh
3、重启终端,更新
$ conda update -n base conda
三、安装tensorflow-gpu及keras
1、创建一个环境,指定python版本
$ conda create -n tf python=3.6
2、激活创建的tf环境:
$ conda activate tf 或者 $ source activate tf
3、在该环境下安装tensorflow-gpu及keras框架:
$ conda install tensorflow-gpu keras
4、查看安装是否成功
$ python
>>import keras
import tensorflow
5、安装其他所需要的库
$ conda install pillow
$ conda install matplotlib
$ conda install scikit-learn
$ conda install graphviz pydot
6、退出该环境
$ conda deactivate 或者 source deactivate
7、移除tf环境
$ conda remove -n tf --all
8、查看GPU运行状况
$ watch -n 0 nvidia-smi