【Windows】在Windows 10安裝CUDA(並使用Python加速 ...

文章推薦指數: 80 %
投票人數:10人

最近在Python執行一些深度學習的程式這些程式的運算很多可以扔到GPU去運算,省下CPU的運算效能,速度也更快不過很多安裝CUDA的教學都是Linux,Windows ... 主頁 文章列表 關於作者 最近在Python執行一些深度學習的程式這些程式的運算很多可以扔到GPU去運算,省下CPU的運算效能,速度也更快不過很多安裝CUDA的教學都是Linux,Windows10用了一天才用好這邊也紀錄一下安裝過程----  操作系統:Windows1064位元NVIDIAGPU:使用CUDA10.2以上的版本====0.安裝前必須先確認自己的顯示卡是否為10.2以上的版本,可以相下兼容,但沒辦法向上雖然這篇文章發佈的時間已經有CUDA11.3的版本了可以安裝了不過後面還要安裝cuDNN,而cuDNN的11.3Windows版本還沒有Windows10版本不同版本沒辦法使用,所以這邊還是選擇10.2的版本,適情況可以選擇適合自己的版本在有安裝顯示卡驅動程式下,右下角的工具列開啟nvidia的控制面版依序點系統資訊->元素->這邊可以確認自己的CUDA版本1.確認沒問題後,可以到nvidiaCUDA的網站下載版本,這裡安裝的是10.2版本https://developer.nvidia.com/cuda-toolkit-archive然後因為10的版本有.1跟.2的Patch修正檔,所以就是都下載下來,依序按主程式然後是2個Patch安裝(在安裝過程中,如果出現visualstudiointegrationfailed而無法正確安裝,請關掉安裝程式列出的程式清單這些程式可能是隱藏執行,需要用工作管理員關閉,安裝程式會列出來,可以注意一下)2.成功安裝後,接著就需要安裝cuDNN,cuDNN是針對學習模型加速的Library類似擴充的函式庫,這邊也可以從nvidia網站下載,不過下載這個必須先申請nvidia的帳號(免費)https://developer.nvidia.com/rdp/cudnn-archive這邊也一樣選擇forCUDA10.2的版本或是看第1步驟的時候是載什麼版本的CUDA下載完解壓縮後,裡面會有bin、include、lib三個資料夾,裡面有一些檔案將檔案放到CUDA的安裝路徑,文章這邊是用預設的路徑:C:\ProgramFiles\NVIDIAGPUComputingToolkit\CUDA\v10.2\到這邊CUDA跟cuDNN就算安裝完畢了,接著是安裝Python的相關環境====3.接著安裝Python的相關環境,這邊使用Anaconda的安裝包https://www.anaconda.com/products/individualAnaconda是一個簡化軟體安裝套件的管理系統,省去要分別載許多軟體的Python安裝包Anaconda2使用Python2.7,Anaconda3則是Python3.7,文章撰寫時使用了Anaconda3的Python3.7版本4.安裝完成後,接著打開AnacondaPrompt來安裝pytorch,這邊先進入pytorch來複製安裝語法https://pytorch.org/進去後稍微往網頁下方拉一下,點選自己的安裝環境需求,下面就會自己跑出語法然後把語法複製貼上到AnacondaPrompt就會開始安裝安裝完成後,最後一步驟就是要啟用GPU加速(CUDA版本必須為10以上)5.打開cmd命令提示字元,輸入pipinstallonnxruntime-gpu這個也安裝完後,Python的程式就可以用GPU運算了!---- Comments 0 Therearenocommentsyet. SECRET Send 更新順 NEWERENTRY 【Unity】自訂UnityApplicationConfigDialog OLDERENTRY 【工作】20210502 同カテゴリ NEWERENTRY 【MachineLearning】用AI還原二次元角色 OLDERENTRY 【Windows】InnoSetup封裝安裝程式並以系統管理員權限執行(新版) About 菲特茶FateTea Client工程師這裡透過工作紀錄一些筆記。

最近的興趣是用Unity做MMD。

[email protected] Plurk(中文) Twitter(日文) Youtube(MMD) bilibili(MMD) Category About(1) Works(9) Diary(3) Unity(35) Unreal(1) MMD(5) SourceTree(8) Program(7) DataBase(2) GameLogic(1) Other(16) NewPost 【Unity】常見MMD4Mecanim相關問題(持續更新)(01/02) 【MachineLearning】用AI還原二次元角色(12/27) 【Unity】MMD4Mecanim介紹(12/21) 【Unity】null與?.(問號點)可能引發的錯誤問題(錯誤:UNT0008)(12/05) 【Unity】UGUI中CullTransprentMesh作用(10/31) 【Unity】UseincrementalGC(10/28) 【Unity】10000次Update()呼叫(10/25) 【Unity】SpriteAtlas(10/22) Message 菲特茶FateTea:【Unity】此版本不符合GooglePlay的64位元版本規定(10/17) 菲特茶FateTea:【Unity】此版本不符合GooglePlay的64位元版本規定(10/16) Neet:【Unity】此版本不符合GooglePlay的64位元版本規定(10/09) Link 管理頁面 將此部落格加到連結 統計人數 Search CLOSE



請為這篇文章評分?