1. 安装CUDA8.0和cudnn 参考 https://www.cnblogs.com/myblog1993/p/6931007.html中的第1、2步
cudnn下载地址:https://download.csdn.net/download/mrkawuyi/10444189
2. 安装Python 3.6
3. 升级pip, 打开cmd输入python.exe -m pip install --upgrade pip
4. 安装TensorFlow,cmd中输入 pip install tensorflow-gpu==1.4.0 如果要获取最新版本输入 pip install tensorflow-gpu(目前最新版本是1.8.0,要求安装CUDA9.0才能使用), 参考 https://www.aliyun.com/jiaocheng/524629.html
5. 等待安装
6. 安装成功就可以测试了,在cmd中输入python,进入调试模式
测试是否安装成功:输入
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
如果出现下列输出表示安装成功!
但是,在测试的过程中,还会遇到一点问题,找不到cudnn64_6.dll,如图
这可能和电脑安装的cudnn版本有关,之前的教程要求安装cudnn后要在系统环境变量中设置Path,我不想设置那么多的path,所以直接将cudnn64_6.dll下载下来放到paython的path下面(我的是C:\Users\Administrator\AppData\Local\Programs\Python\Python36),结果重新调试,TensorFlow居然安装成功了!
详细操作,持续更新中