1、安装cuda Toolkit 和cudnn (百度云可下载,版本需要对应)
2、配置环境变量:
3、安装cudnn(需要拷贝一些dll和lib来进行配置)
4、进入cmd,找到anaconda3的pip路径,用下面的命令来执行,可以卸载cpu版本的tensorflow,安装gpu版本的tensorflow
pip uninstall tensorflow
pip install tensorflow-gpu
完成,tensorflow在训练时候会自动调用gpu来进行计算
折腾了许久,原来是gpu的驱动没有卸载干净,卸载一遍旧的驱动,后来再安装最新的驱动就可以了。
最终成功的是在tensorflow-gpu-1.4.0 + cuda8.0+cudnn V6.0 + win7 x64(最后发现不可行,不知道是不是GPU的问题,总之报错)
只好用这个【 tensorflow-gpu-1.0.0 + cuda8.0+cudnn V5.1 + win7 x64】