TensorFlow-GPU安裝避坑指南

lhrbest發表於2020-02-24

TensorFlow-GPU安裝避坑指南



媳婦安裝 TensorFlow-GPU花了2天的時間還沒有安裝成功,於是,我下定決心幫她安裝一次,遇到了很多問題,現記錄如下。


需要安裝的軟體包括:

1、安裝NVIDIA顯示卡驅動

2、安裝64位的Python 3.6

3、安裝tensorflow-gpu

4、安裝cuda

5、安裝cudnn

6、安裝Microsoft Visual C++ 2015



避坑一: 電腦的顯示卡必須支援NVIDIA,否則不能使用 TensorFlow-GPU。

這個可以用GPU-Z這個軟體來檢測,或直接在電腦裡檢視:


CUDA和CUDNN都是隻支援Nvidia顯示卡的





如果你的電腦不支援CUDA,那就不用進行後邊的安裝步驟了。


下載NVIDIA 驅動程式:



TensorFlow-GPU下載:



避坑二:目前的tensorflow不支援windows32位的python


避坑三:CUDA和CUDNN一定要注意版本的相容性,否則也不能成功安裝。


避坑四: VMware Workstation虛擬機器中不支援GPU的虛擬化。 虛擬機器中的顯示卡是虛擬的,稱為VMware SVGA 3D ,所以,不能使用CUDA ,即無法使用GPU


避坑五: 若安裝CDH版的 NVIDIA驅動 ,則NVIDIA 控制皮膚在Windows 10中需要透過應用商店進行安裝。安裝之後就可以透過英偉達控制皮膚(NVIDIA Control Panel)來檢視當前系統支援的CUDA版本了。或者在安裝 NVIDIA時不要選擇CDH版本,而選擇標準版後安裝後就有 英偉達控制皮膚了。


驅動程式型別需要選擇標準,不要選擇DCH,否則安裝後沒有 NVIDIA 控制皮膚。




----檢測GPU




其它安裝請參考:




NVIDIA、cuda、GPU簡介及版本檢視


一、NVIDIA介紹

NVIDIA,中文名英偉達,是一家人工智慧計算公司,與ATI(後被AMD收購)齊名,專注於打造能夠增強個人和專業計算平臺的人機互動體驗的產品(一家顯示卡廠商),是全球可程式設計圖形處理技術領袖,發明了GPU,重新定義了現代計算機圖形技術,並徹底改變了平行計算。其中 NVIDIA 組織的 NVIDIA GPU 技術大會 (GTC) 是AI和深度學習領域的大會,也是全球範圍的GPU開發者行業聚會。


二、cuda介紹

CUDA(Compute Unified Device Architecture)

1、CUDA是顯示卡廠商NVIDIA推出的運算平臺,是一種通用平行計算架構,該架構使GPU能夠解決複雜的計算問題。 它包含了CUDA指令集架構(ISA)以及GPU內部的平行計算引擎。 開發人員現在可以使用C語言來為CUDA架構編寫程式。

2、隨著顯示卡的發展,GPU越來越強大,而且GPU為顯示影像做了最佳化。在計算上已經超越了通用的CPU。如此強大的晶片如果只是作為顯示卡就太浪費了,因此 NVIDIA 推出CUDA,讓顯示卡可以用於影像計算以外的目的。目前只有G80、G92、G94、G96、GT200、GF100、GF110、GK100、GK104、GK107平臺等NVidia顯示卡才能使用CUDA,工具集的核心是一個C語言編譯器。G80中擁有128個單獨的ALU,因此非常適合平行計算,而且數值計算的速度遠遠優於CPU。

3、CUDA目前支援linux和Windows作業系統。進行CUDA開發需要依次安裝驅動、toolkit、SDK三個軟體(安裝cuda是為使用GPU加速)。

4.檢視cuda版本(安裝環境可能用到)

開啟控制皮膚,選擇NVIDIA控制皮膚

å¨è¿éæ入å¾çæè¿°

選擇幫助下拉選單系統資訊,選擇元件

å¨è¿éæ入å¾çæè¿°

即可看到cuda版本

å¨è¿éæ入å¾çæè¿°


三、GPU介紹

1、顯示卡的處理器稱為圖形處理器(Graphics Processing Unit,GPU),它是顯示卡的“心臟”,與CPU類似,只不過GPU是專為執行復雜的數學和幾何計算而設計的,這些計算是圖形渲染所必需的。

NVIDIAçGPU

2、供應商:英特爾、NVIDIA、AMD等。

3、檢視電腦GPU型號


開啟我的電腦屬性

å¨è¿éæ入å¾çæè¿°

點開裝置管理器

å¨è¿éæ入å¾çæè¿°

點選顯示介面卡



å¨è¿éæ入å¾çæè¿°









About Me

........................................................................................................................

● 本文作者:小麥苗,部分內容整理自網路,若有侵權請聯絡小麥苗刪除

● 本文在itpub、部落格園、CSDN和個人微 信公眾號( xiaomaimiaolhr)上有同步更新

● 本文itpub地址: http://blog.itpub.net/26736162

● 本文部落格園地址: http://www.cnblogs.com/lhrbest

● 本文CSDN地址: https://blog.csdn.net/lihuarongaini

● 本文pdf版、個人簡介及小麥苗雲盤地址: http://blog.itpub.net/26736162/viewspace-1624453/

● 資料庫筆試面試題庫及解答: http://blog.itpub.net/26736162/viewspace-2134706/

● DBA寶典今日頭條號地址:

........................................................................................................................

● QQ群號: 230161599 、618766405

● 微 信群:可加我微 信,我拉大家進群,非誠勿擾

● 聯絡我請加QQ好友 646634621 ,註明新增緣由

● 於 2020-02-01 06:00 ~ 2020-02-31 24:00 在西安完成

● 最新修改時間:2020-02-01 06:00 ~ 2020-02-31 24:00

● 文章內容來源於小麥苗的學習筆記,部分整理自網路,若有侵權或不當之處還請諒解

● 版權所有,歡迎分享本文,轉載請保留出處

........................................................................................................................

小麥苗的微店

小麥苗出版的資料庫類叢書http://blog.itpub.net/26736162/viewspace-2142121/

小麥苗OCP、OCM、高可用網路班http://blog.itpub.net/26736162/viewspace-2148098/

小麥苗騰訊課堂主頁https://lhr.ke.qq.com/

........................................................................................................................

使用 微 信客戶端掃描下面的二維碼來關注小麥苗的微 信公眾號( xiaomaimiaolhr)及QQ群(DBA寶典)、新增小麥苗微 信, 學習最實用的資料庫技術。

........................................................................................................................

歡迎與我聯絡

 

 



來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/26736162/viewspace-2676960/,如需轉載,請註明出處,否則將追究法律責任。

相關文章