在深度學(xué)習(xí)訓(xùn)練的時(shí)候使用GPU而不是CPU我想已經(jīng)是不爭的事實(shí)了，雖然MX150并不在下面的官網(wǎng)列表，但是其實(shí)MX150也是支持CUDA的。

讓客戶滿意是我們工作的目標(biāo)，不斷超越客戶的期望值來自于我們對這個(gè)行業(yè)的熱愛。我們立志把好的技術(shù)通過有效、簡單的方式提供給客戶，將通過不懈努力成為客戶在信息化領(lǐng)域值得信任、有價(jià)值的長期合作伙伴，公司提供的服務(wù)項(xiàng)目有：國際域名空間、雅安服務(wù)器托管、營銷軟件、網(wǎng)站建設(shè)、舞陽網(wǎng)站維護(hù)、網(wǎng)站推廣。

https://developer.nvidia.com/cuda-gpus

環(huán)境

小米筆記本Air 13

• OS: win10
• CPU: I7 7500U
• GPU: MX150
• GPU驅(qū)動: 425.25
• tensorflow: 1.13.1
• tensorflow-gpu: 1.13.1
• visual studio: 2019

安裝TensorFlow

現(xiàn)在TensorFlow的whl文件已經(jīng)打包的非常好了，基本上是可以安裝上的，但是如果沒有CUDA之類的驅(qū)動的話，在導(dǎo)入tensorflow的時(shí)候會報(bào)錯(cuò)。

pip install tensorflow tensorflow-gpu

安裝依賴

為了裝上英偉達(dá)的CUDA套件還需要安裝visual studio, 因?yàn)閣indows的相關(guān)編譯環(huán)境跟visual studio綁在了一起，即使你只想裝其中一部分，還是得裝上visual studio

安裝visual studio

而windows的安裝程序不會太難，就是下一步，下一步。

安裝 cuda toolkits

CUDA toolkits 10.0

https://developer.nvidia.com/cuda-zone

默認(rèn)安裝即可，下一步下一步。

安裝cudnn

cudnn 7.6.0.64

https://developer.nvidia.com/cudnn

注意千萬不要貪最新的版本，tensorflow官方不一定支持!

cudnn解壓后放在C:\tools\cuda

最后加入環(huán)境變量:

C:\tools\cuda\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64

官方說明如下:

軟件要求

必須在系統(tǒng)中安裝以下 NVIDIA? 軟件：
NVIDIA? GPU 驅(qū)動程序 - CUDA 10.0 需要 410.x 或更高版本。

CUDA? 工具包 - TensorFlow 支持 CUDA 10.0（TensorFlow 1.13.0 及更高版本）

CUDA 工具包附帶的 CUPTI。
cuDNN SDK（7.4.1 及更高版本）
（可選）

TensorRT 5.0，可縮短在某些模型上進(jìn)行推斷的延遲并提高吞吐量。

參考頁面:
https://www.tensorflow.org/install/gpu

各個(gè)版本的兼容測試情況
https://www.tensorflow.org/install/source#linux

安裝參考:

https://towardsdatascience.com/installing-tensorflow-with-cuda-cudnn-and-gpu-support-on-windows-10-60693e46e781

https://medium.com/@johnnyliao/%E5%9C%A8nvidia-mx150%E7%9A%84win10%E5%AE%89%E8%A3%9Dcuda-toolkit-cudnn-python-anaconda-and-tensorflow-91d4c447b60e

當(dāng)前標(biāo)題：讓tersorflow利用GPU飛起來-MX150
當(dāng)前路徑：http://aaarwkj.com/article10/jjpjdo.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供自適應(yīng)網(wǎng)站、網(wǎng)站設(shè)計(jì)、靜態(tài)網(wǎng)站、微信小程序、網(wǎng)站收錄、外貿(mào)建站

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時(shí)需注明來源： 創(chuàng)新互聯(lián)