Windows下 python 3 + tensorflow-gpu安装配置

  • 首先要安装vs2015以上版本
  • pip install tensorflow
  • 如果下载速度太慢, 可以换国内源
  • pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/ tensorflow
  • 快的一批, 从几十k, 变成了几M.
  • 尝试在IDLE中 import tensorflow
  • 仔细看报错最后几行, 会说缺少动态库, 然后让你安装CUDA xx.x, 后面给你一个链接, 不要用链接, 直接百度CUDA xx.x安装, 去官网下载, 推荐下载本地版, 我这边是比较快. 人家说要啥版本就装啥版本, 要10.0, 你就不能装10.1
  • 安装CUDA的时候, 推荐全部默认, 看不懂就不要乱改了.
  • 环境变量配置部分看这里https://blog.csdn.net/u013165921/article/details/77891913
  • 然后去下载CUDNN, 要下载与CUDA对应的版本, 注意, 要在注册登录之后才能下载
  • 下载完之后解压, 会有三个文件夹, 把里面的东西分别复制到CUDA路径下的对应位置
  • 最后再在IDLE里import tensorflow