配置tensorflow-gpu

最近从theano 转到tensorflow上来，前两天一直在配环境，希望这篇博客可以帮助大家配置tensorflow-gpu

我的那台服务器忘记安装nvidia驱动直接就配好了tensorflow-cpu，首先将tensorflow卸载。

1 安装nvidia驱动

2 安装cuda-8.0,这里我曾经尝试着安装cuda9.1.发现conda install tensorflow-gpu 这个命令安装的tensorflow不能使用cuda9.1.所以只能改用cuda-8.0

3安装cudnn，这个可以参照官网的安装教程

4 配置环境变量

5 在控制台输入 conda install tensorflow-gpu 即可

注意：我这里是在 source activate (conda 环境名) 之后，在环境里安装的tensorflow-gpu

之后便可以运行了。如还有问题可以参照官网： https://www.tensorflow.org/install/install_linux