1. 程式人生 > >Py之TF/Cuda/Cudnn:Win10下安裝深度學習框架Tensorflow+Cuda+Cudnn最簡單最快捷最詳細攻略

Py之TF/Cuda/Cudnn:Win10下安裝深度學習框架Tensorflow+Cuda+Cudnn最簡單最快捷最詳細攻略

本人在Win10下安裝深度學習框架Tensorflow,安裝之前各種谷歌,各種百度,各種國內外資料,做了充分準備,

第一步:先洗臉

第二步:照照鏡子

第三步:哈哈,進入正題

因為現在電腦比較新,差不多都支援GPU運算,所以我以GPU下安裝Tensorflow

Win+R執行命令視窗,然後cmd進入dos環境,輸入命令
pip install tensorflow-gpu

 pip3 install --upgrade tensorflow-gpu

一步,二步,三步,向前走……



四步,五步接著走啊……

說走咱就走啊,大河向東流啊

第四步:哈哈,繼續安裝Cuda和Cudann

先解釋一下

Cuda:NVIDIA Cuda是用於深度神經網路的GPU加速庫。它強調效能、易用性和低記憶體開銷。

Cudnn:NVIDIA Cudnn可以整合到更高級別的機器學習框架中。能夠程式設計使用和控制GPU進行相應的邏輯工作,比如進行神經網路學習,這還不夠,因為不夠方便,如果有專門的工具包或者演算法包,直接呼叫就好了,比如像JDK。於是有了CUDNN。

Cudnn支援的演算法

  1. 卷積操作、相關操作的前向和後向過程。
  2. pooling的前向後向過程
  3. softmax的前向後向過程
  4. 啟用函式的前向後向過程 
    • ReLU
    • sigmoid
    • TANH
  5. Tensor轉換函式,其中一個Tensor就是一個四維的向量。

第五步:安裝Cuda

第一個視窗路徑是軟體安裝路徑最好不要預設安裝在C盤,改到C盤以外的F盤;第二個視窗安裝路徑中配置環境路徑預設在C盤,其他路徑最好放置到C盤以外的F盤。當然,如果你不懂配置路徑的話,就預設吧!

Cuda安裝檔案,百度就能找到,長的是這個模樣


Cuda安裝過程


dos內執行命令nvcc -V,記住,是在dos內執行,驗證是否安裝成功,當出現以下內容即是安裝成功!


第六步:下載並配置Cudnn

然後將下邊三個資料夾,複製到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0內即可



第七步:進行測試是否安裝成功

開啟Python3,輸入import tensorflow,回車即可,以下即為匯入成功,哈哈,可以進行深度學習啦!


安裝中有任何問題,前來諮詢,和你華山論劍!哈哈,繼續學習吧!