TensorFlow windows安装

[toc]

为什么

要做毕设了,然后本专业的人都走了,很无聊呀。想做点新鲜的事情
我的环境是 e3-1231 v3 + gtx970

超级坑的

现在在windows上的cuda9很坑比,你的tensorflow 最好安装cuda8.0
cudnn 要6.0的 7.0不行
看清楚是什么错误:ImportError: Could not find ‘cudnn64_6.dll’. TensorFlow requires that this DLL be installed

windows 安装必知

不久前支持安装windows的
官网安装文档:
安装时可以选择自定义安装,只要安装cuda就行了 那个其他驱动不需要安装

  1. 首先要用GPU的话只能使用英伟达的CUDA,然后要下一个CUDA Toolkit
  2. 下载安装python 我自己装的是3.6
  3. 使用pip安装tensorflow gpu版:pip3 install --upgrade tensorflow-gpu
    • 可能是这个翻墙比较快吧,外加校园网,一会就安装完成了
  4. 安装cudnncudnn6.0 要注册没办法呀

如何卸载cuda9 和安装cuda8(mad坑呀)

  • 使用windows自带的程序卸载,把带有9.1 和 cuda的都给卸载了。不用考虑顺序。选择按照安装时间看
  • 由于安装cuda时一般还安装了自带的驱动。所以不要把驱动给卸载了

结果


mad 终于完成了
cuda8.0 + cudnn6.0 绝配


评论

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注