准备工作

  安装前请确保自己的显卡支持gpu加速,支持加速的gpu型号可在下面的链接中查询。

  https://www.geforce.com/hardware/technology/cuda/supported-gpus

SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。

一、安装anaconda

  1.   推荐从清华镜像站下载,速度快。
  2.   根据系统选择安装包,现(2018.11.21)推荐Anaconda3-5.3.0及更高版本,因为Tensorflow暂不支持Python3.7。
  3.   安装时勾选Add Anaconda to my PATH environment variable

 

二、安装cuda及cudnn

  1. 下载安装cuda,https://developer.nvidia.com/cuda-downloads
  2. 下载cudnn,https://developer.nvidia.com/cudnn ,(下载cudnn需要注册账号,请根据cuda版本选择相对应cudnn)
  3. 解压cudnn,将解压出来的三个文件夹下面的文件移动到对应的cuda文件夹下(cuda默认安装路径为 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0)    Windows10系统tensorflow-gpu安装 人工智能 第1张
  4. 将cuda的lib和include文件夹路径添加到环境变量
  5. Windows10系统tensorflow-gpu安装 人工智能 第2张

三、下载安装tensorflow-gpu

  打开命令行,执行下面命令。

pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple  //同样使用清华镜像源,下载速度快

  测试一下

  Windows10系统tensorflow-gpu安装 人工智能 第3张

  完成

 

扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄