配置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,这个可以参照官网的安装教程

配置tensorflow-gpu

4 配置环境变量

配置tensorflow-gpu

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

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

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