windows安裝tensorflow簡單直接的方法(win10+pycharm+tensorflow-gpu1.7+cuda9.1+cudnn7.1)

標籤：不能   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 tfh = 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

　　在ｃｍｄ輸入nvcc -V測試安裝是否成功。

四，回到pycharm,運行程式

　　

安裝成功！！！

五，目前tensorflow最高支援CUDA9.0如果要使用CUDA9.1:

　　安利大牛用CUDA９.１自己編譯的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)