NVIDIA/cuda-samples

解決的問題

此倉儲為使用 NVIDIA CUDA Toolkit 的開發者提供全面的參考實作與範例集合。協助開發者學習如何使用 CUDA 功能、實作 GPU 加速演算法，並驗證其 CUDA 環境是否正確設定。

作法說明

專案分為 C++ 與 Python 範例。C++ 範例使用 CMake 建置，可針對特定 GPU 架構（SM 版本）進行編譯。Python 範例透過 CUDA Python 套件使用 cuda.core API，並與 NumPy 及 CuPy 整合以處理記憶體與程式啟動。

適用對象

適合需要 GPU 程式設計實用範例的 CUDA 開發者，從初學者學習基礎向量加法，到進階使用者實作分散式運算或與 PyTorch 及 TensorFlow 的框架互通。

主要特色

• 多語言支援：包含 C++ 與 Python 範例。
• 廣泛平台支援：支援 Linux、Windows，以及 Tegra 與 QNX 平台的跨平台編譯。
• 全面涵蓋：涵蓋歸約、直方圖、FFT、串流重疊與多 GPU 對等通訊等核心概念。
• 自動化測試：包含 run_tests.py 指令碼，可作為環境健全性檢查執行所有範例。
• 框架互通：提供與 PyTorch 及 TensorFlow 集成的特定範例。