【Win7 x64】+【annaconda3】+ 【python3.5.2】+【tensorflow-gpu】 [最终配置 gtx 940mx + Cuda8.0+cudnn v5.1 + tensorflow-gpu1.0.0 ]

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】

原文地址:https://www.cnblogs.com/jerrybaby/p/7764560.html