首页 > Windows开发 > 详细

Windows10系统tensorflow-gpu安装

时间:2018-11-21 19:04:45      阅读:223      评论:0      收藏:0      [点我收藏+]

 

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

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

一、安装anaconda

  1.   推荐从清华镜像站下载,速度快。
  2.   根据自己系统选择安装包,现(18.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)    技术分享图片
  4. 将cuda的lib和include文件夹添加到环境变量

三、下载安装tensorflow-gpu

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

pip install tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

Windows10系统tensorflow-gpu安装

原文:https://www.cnblogs.com/clearlove7777777/p/9996918.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!