在 Windows 11 與 Ubuntu Linux 24.04 的命令列檢查 PyTorch 是否看得到 CUDA GPU。程式會印出 torch 版本、CUDA 是否可用、裝置數量、目前使用的裝置編號,以及每一張裝置的名稱——在開始真正的專案之前,這是確認機器是否已具備 GPU 訓練能力最快的方法。

 

[資料夾結構]

  • CLAUDE.md / AGENT.md:程式規格書(內容相同)。
  • CHANGELOG.md:變更紀錄。
  • copyright.txt:版權聲明。
  • src/:測試程式、啟動腳本與本說明文件。
  • venv/:Python 虛擬環境,由步驟 1 建立。

[執行環境需求]

  • 作業系統:Windows 11 或 Ubuntu Linux 24.04。
  • Python 3 與 pip。
  • Python 套件:torch,由步驟 1 自動安裝。
  • 若要測到 GPU:需要 NVIDIA 顯示卡與最新的驅動程式。沒有 GPU 的電腦一樣可以執行,只會回報找不到裝置。
  • 第一次下載 torch 時需要網路。

[操作步驟及說明]

步驟 1:建立虛擬環境

建立虛擬環境並安裝 torch。這是 torch 專案,在有 GPU 的電腦上請使用 01_venv_create_gpu 腳本,它會安裝 CUDA 12.8 版本的 torch——一般的 01_venv_create 安裝的是 CPU 版,執行結果一定會顯示 CUDA 不可用。

 

Windows 11,GPU:

src\01_venv_create_gpu.bat

Windows 11,CPU:

src\01_venv_create.bat

Ubuntu Linux 24.04,GPU:

bash src/01_venv_create_gpu.sh

Ubuntu Linux 24.04,CPU:

bash src/01_venv_create.sh

步驟 2:啟動虛擬環境

在目前的命令列視窗啟動虛擬環境。Linux 必須用 source 執行,啟動狀態才會在腳本結束後繼續保留。

 

Windows 11:

src\02_venv_activate.bat

Ubuntu Linux 24.04:

source src/02_venv_activate.sh

步驟 3:執行 CUDA 檢查

執行檢查程式。報告會列出 torch 版本、CUDA 是否可用、裝置數量、目前裝置的編號,以及每一張裝置的名稱。

 

Windows 11:

src\03_cuda_test.bat

Ubuntu Linux 24.04:

bash src/03_cuda_test.sh

步驟 4:移除虛擬環境

使用完畢後,刪除 venv/ 資料夾以釋放磁碟空間。需要時再重新執行步驟 1 即可。

 

Windows 11:

src\99_venv_clear.bat

Ubuntu Linux 24.04:

bash src/99_venv_clear.sh

[結果判讀]

  • 在沒有 GPU 的電腦上,或安裝的是 CPU 版 torch,程式會回報 0 個裝置並正常結束——查詢裝置的部分有加上防護,不會當掉。
  • 明明裝了 NVIDIA 顯示卡卻顯示 CUDA 不可用,幾乎都是因為安裝到 CPU 版的 torch,請改用 01_venv_create_gpu 重建虛擬環境。
  • 裝置數量是 torch 可以使用的 GPU 數量;目前裝置是程式預設會選用的編號,除非程式另外指定,否則就是 0。
  • torch 回報的 CUDA 版本是編譯時所使用的版本,不必和 nvidia-smi 顯示的版本一致——只要驅動程式夠新即可。

這個 SDK 內建於 MYAI Studio 人工智慧開發工具。

 

單獨購買價格:5,000元,永久授權,單一 SDK 授權,單機授權,一次下載,無更新服務,一年電子郵件範例諮詢服務。如果是軟體客製化問題,請購買我們的客製化顧問服務點數。