玩AI模型,GPU算力是基礎,首先得了解自己電腦的 CUDA 和 GPU 信息,并安裝相對應的軟件、進行正確的配置。 要查看電腦的 CUDA 和 GPU 信息,可以通過命令行查看 查看 GPU 信息,打開命令提示符或 PowerShell,然后輸入以下命令: nvidia-smi ![]() 這個命令會顯示電腦的 GPU 型號、驅動程序版本以及 CUDA 版本。根據 nvidia-smi 輸出的信息, GPU 是 NVIDIA GeForce,并且當前安裝的 CUDA 版本是 11.4。 查看 CUDA Toolkit 版本,打開命令提示符或 PowerShell,然后輸入以下命令: nvcc --version 這個命令會顯示安裝的 CUDA Toolkit 的版本信息。 如果沒有安裝 nvcc,你可能需要先安裝 CUDA Toolkit。要在你的系統上安裝 CUDA Toolkit,可以按照以下步驟進行: 下載 CUDA Toolkit 安裝包: https://developer.nvidia.com/cuda-toolkit-archive 訪問 NVIDIA CUDA Toolkit 下載頁面。 選擇合適的版本(例如 CUDA 11.4)和你的操作系統。 ![]() 下載對應的安裝包。 ![]() 安裝 CUDA Toolkit:運行下載的安裝包并按照提示完成安裝。 在安裝完成后,需要將 CUDA 的 bin 目錄和 libnvvp 目錄添加到系統的環境變量中。 打開“系統屬性” -> “高級系統設置” -> “環境變量”。 在“系統變量”部分,找到 Path,點擊“編輯”。 添加以下路徑(根據你的安裝路徑進行調整): 復制代碼 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\libnvvp 驗證安裝: ![]() 打開命令提示符,輸入以下命令來驗證 CUDA Toolkit 是否安裝成功: nvcc --version 你應該能看到 CUDA 編譯器的版本信息。 ![]() 最后,要安裝 cuDNN 訪問 NVIDIA cuDNN 下載頁面。 https://developer.nvidia.com/cudnn ![]() 登錄并下載與 CUDA 版本匹配的 cuDNN 文件。 ![]() 然后安裝 cuDNN 。 |
|