[Python]安裝CUDA + cuDNN

[Python]安裝CUDA + cuDNN

更新於 發佈於 閱讀時間約 5 分鐘

要使用 CUDA(Compute Unified Device Architecture)來加速計算,首先需要在你的系統上設置和安裝相關的工具。CUDA 是由 NVIDIA 開發的平行計算框架,用於加速大量數據的運算,尤其在圖像處理、機器學習、科學計算等領域很有應用。

可以參考官方的安裝方式

以下是使用 CUDA 的具體步驟和方法:

1. 確認系統支援 CUDA

確保你的系統配備了 NVIDIA 的 GPU,且 GPU 支援 CUDA。

你可以查詢 NVIDIA 的官方網站來確認你的 GPU 是否支援 CUDA。通常較新的 NVIDIA 顯卡都支援 CUDA。

或從NVIDIA控制面板→系統資訊→元素這可觀看

raw-image



2. 安裝 NVIDIA 驅動程式

確保你的系統已經安裝了適合的 NVIDIA 顯示驅動程式。你可以從 NVIDIA 的官方驅動下載頁面來獲取正確的驅動程式。


3. 安裝 CUDA 工具包

前往 NVIDIA 的 CUDA Toolkit 下載頁面 並下載對應你操作系統版本的 CUDA Toolkit。CUDA Toolkit 包含了 CUDA 的開發工具和編譯器。

raw-image

安裝步驟:

  • 下載對應的 CUDA Toolkit 安裝包。
  • 根據官方提供的指引進行安裝,過程中會包含編譯器、標頭文件、庫等所需工具。

我是使用線上安裝版的步驟

使用 CUDA 的線上安裝版(exe network installer) 時,安裝程序會自動連接到 NVIDIA 的伺服器,並根據你當前的系統環境來檢查並下載適合的驅動程序和 CUDA 工具包。因此,它確實會幫你檢查並安裝適合的 CUDA 版本

raw-image
raw-image
raw-image
raw-image
raw-image
raw-image
  • 安裝完會自動設置好環境變數,將 CUDA 添加到 PATH中。
raw-image

4. 安裝 cuDNN

如果你打算在深度學習框架中使用 CUDA,加速神經網絡的計算,可以安裝 cuDNN(CUDA Deep Neural Network library)。你可以從 NVIDIA cuDNN 官方下載頁面 下載對應的版本,並將它解壓到 CUDA Toolkit 對應的目錄中。

需註冊會員才能下載

下載完 cuDNN 放置位置

當你下載並解壓 cuDNN 後,需要將它的內容放到已安裝的 CUDA 工具包對應的目錄中。具體操作如下:

  • 解壓縮 cuDNN 壓縮包,你會看到三個主要文件夾:binincludelib
  • 將這些文件複製到對應的 CUDA 目錄
    • bin 文件夾(動態鏈接庫 .dll 文件): 將解壓縮的 bin 目錄下的 .dll 文件放入:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\bin
raw-image
    • include 文件夾(頭文件 .h 文件): 將解壓縮的 include 目錄下的 cudnn.h 放入:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\include
raw-image
    • lib 文件夾(靜態庫和動態鏈接庫 .lib 文件): 將解壓縮的 lib\x64 目錄下的 .lib 文件放入:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6\lib\x64
raw-image

這樣做就完成了 cuDNN 與 CUDA 工具包的集成。

5. 確認 CUDA 安裝是否成功

安裝完 CUDA Toolkit 後,確認是否安裝成功。你可以使用以下命令來檢查 CUDA 安裝和 GPU 狀態:

nvcc --version

這會顯示 CUDA 編譯器的版本,如果安裝正確,你會看到 CUDA 版本信息。

raw-image

. Cuda compilation tools, release 12.6, V12.6.77

這部分表示:

  • CUDA 工具鏈版本:你使用的 CUDA 工具是 12.6 版
  • V12.6.77 是這個工具鏈的具體版本號。






avatar-img
螃蟹_crab的沙龍
138會員
244內容數
本業是影像辨識軟體開發,閒暇時間進修AI相關內容,將學習到的內容寫成文章分享。
留言
avatar-img
留言分享你的想法!
螃蟹_crab的沙龍 的其他內容
 AGV最佳路徑分析(自動導引車最佳路徑分析)時,目的是讓 AGV(自動導引車)在工廠、倉庫或其他環境中,找到最短、最有效率且安全的路徑完成任務。 什麼是AGV? AGV 是一種自動駕駛的小車,可以運送物品,類似於會自己走路的小機器人。比如在超市運送牛奶,或者在工廠裡搬運零件。 什麼是最佳
Dijkstra 演算法實作 Dijkstra 是用來找到從一個起點到其他所有點的最短路徑的演算法,適合靜態地圖。 實作步驟 定義地圖:使用圖(Graph)表示地圖。 初始化距離:起點距離設為 0,其餘點設為無限大。 遍歷節點:逐步找出距離最短的節點,更新相鄰節點的距離。 結束條件:所有
A* (A-Star)演算法實作 A* (A-Star)是一種啟發式搜尋演算法,除了最短距離,還會考慮目標的估計距離(啟發式函數)。 實作步驟 定義地圖和啟發式函數。 初始化開放節點(Open Set)和已訪問節點(Closed Set)。 計算代價函數 f(n) = g(n) + h(n
 AGV最佳路徑分析(自動導引車最佳路徑分析)時,目的是讓 AGV(自動導引車)在工廠、倉庫或其他環境中,找到最短、最有效率且安全的路徑完成任務。 什麼是AGV? AGV 是一種自動駕駛的小車,可以運送物品,類似於會自己走路的小機器人。比如在超市運送牛奶,或者在工廠裡搬運零件。 什麼是最佳
Dijkstra 演算法實作 Dijkstra 是用來找到從一個起點到其他所有點的最短路徑的演算法,適合靜態地圖。 實作步驟 定義地圖:使用圖(Graph)表示地圖。 初始化距離:起點距離設為 0,其餘點設為無限大。 遍歷節點:逐步找出距離最短的節點,更新相鄰節點的距離。 結束條件:所有
A* (A-Star)演算法實作 A* (A-Star)是一種啟發式搜尋演算法,除了最短距離,還會考慮目標的估計距離(啟發式函數)。 實作步驟 定義地圖和啟發式函數。 初始化開放節點(Open Set)和已訪問節點(Closed Set)。 計算代價函數 f(n) = g(n) + h(n