碼迷,www.tparu.icu
吉利平特名人堂 > Windows程序 > 詳細

平特乾坤卦利港丰:windows安裝tensorflow簡單直接的方法(win10+pycharm+tensorflow-gpu1.7+cuda9.1+cudnn7.1)

時間:2018-04-06 20:17:53      閱讀:6668      評論:0      收藏:0      [點我收藏+]

吉利平特名人堂 www.tparu.icu 標簽:不能   baidu   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)

標簽:不能   baidu   import   直接   需要   www   項目   大牛   pip   

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

(2)
(0)
   
舉報
評論 一句話評論(0
0條  
登錄后才能評論!
? 2014 吉利平特名人堂 版權所有 京ICP備13008772號-2
迷上了代碼!
三分时时人免费计划 老重庆时时彩 优博时时彩平台 11选5前2直选技巧 差旅费报销单电子版 bet007足球即时比分球探网 快乐时时是国家 双色球免费计划软件 幸运飞艇双面玩法怎么玩 布鲁日 极速时时开奖视频 天天彩票助赢计划软件下载 手机投注违法吗 时时彩开奖 北京pk10做号安卓版 3d1胆拖7码直选多少钱