在win7搭建tensorflow-gpu

  • 环境
    win7的64位专业版
    anoconda最新版(不用在意python版本,之后创建新的虚拟环境时直接指定新的python版本即可)
    vs 2015 + cuda9.0 + cudnn7 + tensorflow-gpu1.7.0 + python3.6.0

  • 安装过程:参考安装过程

  • 安装前注意:
    确保电脑上有独立的英伟达显卡且安装了相应的驱动,(查看显卡型号:右击计算机-设备管理器-显示适配器)。如下图
    在win7搭建tensorflow-gpu
    然后,去官网查看cuda是否支持该显卡型号且计算能力最好在3.0及以上,才有继续安装的必要性

  • 安装过程的补充:
    安装tensorflow-gpu时,我采用的是在tensorflow虚拟环境下直接:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.7.0

  • 安装后可能遇到的问题:
    1、若安装好后报错,可能是cuda安装好后没有设置环境变量,具体百度
    2、import tensorflow时报错: ImportError: DLL load failed: 找不到指定的程序,参考解决方法+解决方法
    。我的解决方法是把protobuf换成3.9.2版本(我的tensorflow-gpu是1.7.0)。若还不行,可参考这个比较根本的解决方法(未实践):解决方法