Windows运维之windows安装tensorflow简单直接的方法
小标 2018-08-29 来源 : 阅读 1538 评论 0

摘要:本文主要向大家介绍了Windows运维之windows安装tensorflow简单直接的方法,通过具体的内容向大家展现,希望对大家学习Windows运维有所帮助。

本文主要向大家介绍了Windows运维之windows安装tensorflow简单直接的方法,通过具体的内容向大家展现,希望对大家学习Windows运维有所帮助。

安装tensorflow-gpu环境需要:python环境,tensorflow-gpu包,cuda,cudnn
一,安装python,pip3直接到官网下载就好了,下载并安装你喜欢的版本
       https://www.python.org/
     提示:安装最后一步时记得勾选添加环境变量
      在cmd输入pip3测试pip3能否使用,不能使用的话,手动打开python安装的路径,找到pip3文件,将路径加入环境变量
      
二,安装tensorflow-gpu
      使用pip3安装即可:pip3 install tensorflow-gpu  
       
     等待下载完成后,会提示安装了那些包,提示安装成功。
  
三,安装CUDA,cudnn
  安装好tensorflow-gpu后,打开你安装好的pycharm(或者你喜欢的IDE)
  输入一段测试用的代码

import tensorflow as tf
h = tf.constant("hell,TF")
sess=tf.Session()
print(sess.run(h))

  运行,这时会有报错信息,报错信息里面会有CUDA的版本信息,按照提示下载对应版本就好(报错信息有链接)。
  下载安装CUDA后,下载对应的cudnn (https://developer.nvidia.com/rdp/cudnn-download)
  CUDA的安装路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1
  将下载好的cudnn文件解压出来,打开CUDA安装路径。
  将cudnn的每个文件夹里的文件复制到CUDA文件夹对应的文件夹中。OK
  在cmd输入nvcc -V测试安装是否成功。
四,回到pycharm,运行程序
  
安装成功!!!
五,目前tensorflow最高支持CUDA9.0如果要使用CUDA9.1:
  安利大牛用CUDA9.1自己编译的whl文件。
  https://github.com/fo40225/tensorflow-windows-wheel
  在项目中找到对应的版本下载就好了。
  将之前的tensorflow卸载:pip3 uninstall tensorflow-gpu
  重新安装:pip3 install 文件名.whl

本文由职坐标整理并发布,希望对同学们有所帮助。了解更多详情请关注系统运维windows频道!

本文由 @小标 发布于职坐标。未经许可,禁止转载。
喜欢 | 0 不喜欢 | 0
看完这篇文章有何感觉?已经有0人表态,0%的人喜欢 快给朋友分享吧~
评论(0)
后参与评论

您输入的评论内容中包含违禁敏感词

我知道了

助您圆梦职场 匹配合适岗位
验证码手机号,获得海同独家IT培训资料
选择就业方向:
人工智能物联网
大数据开发/分析
人工智能Python
Java全栈开发
WEB前端+H5

请输入正确的手机号码

请输入正确的验证码

获取验证码

您今天的短信下发次数太多了,明天再试试吧!

提交

我们会在第一时间安排职业规划师联系您!

您也可以联系我们的职业规划师咨询:

小职老师的微信号:z_zhizuobiao
小职老师的微信号:z_zhizuobiao

版权所有 职坐标-一站式AI+学习就业服务平台 沪ICP备13042190号-4
上海海同信息科技有限公司 Copyright ©2015 www.zhizuobiao.com,All Rights Reserved.
 沪公网安备 31011502005948号    

©2015 www.zhizuobiao.com All Rights Reserved