码迷,mamicode.com
首页 > Windows法式榜样 > 详细

windows装置tensorflow简单直接的办法(win10+pycharm+tensorflow-gpu1.7+cuda9.1+cudnn7.1)

时间:2018-04-06 20:17:53      浏览:9593      评论:0      收藏:0      [点我收藏+]

标签:不克不及   百度   import   直接   须要   www   项目   大年夜牛   pip   

装置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装置tensorflow简单直接的办法(win10+pycharm+tensorflow-gpu1.7+cuda9.1+cudnn7.1)

标签:不克不及   百度   import   直接   须要   www   项目   大年夜牛   pip   

原文地址:https://www.cnblogs.com/MrRead/p/8728569.html

(2)
(0)
   
告发
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权一切 京ICP备13008772号-2
迷上了代码!