Py之TF/Cuda/Cudnn:Win10下安裝深度學習框架Tensorflow+Cuda+Cudnn最簡單最快捷最詳細攻略
阿新 • • 發佈:2019-02-08
本人在Win10下安裝深度學習框架Tensorflow,安裝之前各種谷歌,各種百度,各種國內外資料,做了充分準備,
第一步:先洗臉
第二步:照照鏡子
第三步:哈哈,進入正題
因為現在電腦比較新,差不多都支援GPU運算,所以我以GPU下安裝TensorflowWin+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支援的演算法
- 卷積操作、相關操作的前向和後向過程。
- pooling的前向後向過程
- softmax的前向後向過程
- 啟用函式的前向後向過程
- ReLU
- sigmoid
- TANH
- 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,回車即可,以下即為匯入成功,哈哈,可以進行深度學習啦!
安裝中有任何問題,前來諮詢,和你華山論劍!哈哈,繼續學習吧!