入門指南：在本地安裝 PyTorch 或在受支援的雲平臺上即刻啟動
加入 PyTorch 基金會
作為 PyTorch 基金會的成員，您將獲得相關資源，協助維護穩定、安全且持久的程式碼庫。您可以協作開展培訓、本地和區域活動、開源開發者工具開發、學術研究，並提供指南幫助新使用者和貢獻者獲得高效的體驗。
主要特性與功能
生產就緒
利用 TorchScript 在即時模式（eager mode）和圖模式（graph mode）之間無縫切換，並使用 TorchServe 加速邁向生產環境的步伐。
分散式訓練
torch.distributed 後端支援在研究和生產環境中實現可擴充套件的分散式訓練及效能最佳化。
強大的生態系統
豐富的工具和庫生態系統擴充套件了 PyTorch 的能力，支援計算機視覺、NLP 等領域的開發。
雲端支援
PyTorch 在各大主流雲平臺上均得到完善支援，為您提供順暢的開發體驗和便捷的擴充套件能力。
安裝 PyTorch
選擇您的偏好並執行安裝命令。“Stable”（穩定版）代表當前經過充分測試和支援的 PyTorch 版本，適合大多數使用者。“Preview”（預覽版）適用於希望嚐鮮最新的、尚在測試中的每日構建版本。請確保您已滿足以下先決條件（例如 numpy），具體取決於您的軟體包管理器。您也可以安裝舊版本的 PyTorch。注意：LibTorch 僅適用於 C++。
注意：最新穩定版 PyTorch 需要 Python 3.10 或更高版本。
PyTorch 構建版本
您的作業系統
安裝包
程式語言
計算平臺
執行此命令
PyTorch 構建版本
穩定版 (2.7.0)
預覽版 (每日構建)
您的作業系統
Linux
Mac
Windows
安裝包
Pip
LibTorch
原始碼
程式語言
Python
C++ / Java
計算平臺
CUDA 11.8
CUDA 12.6
CUDA 12.8
ROCm 6.3
CPU
執行此命令
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
雲合作伙伴快速入門
透過主流雲平臺和機器學習服務，快速部署並執行 PyTorch。
生態系統
精選專案
探索豐富的庫、工具及更多專案，助力您的開發工作。
Captum
Captum（拉丁語意為“理解”）是一個基於 PyTorch 構建的開源、可擴充套件模型可解釋性庫。
PyTorch Geometric
PyTorch Geometric 是一個用於不規則輸入資料（如圖形、點雲和流形）深度學習的庫。
skorch
skorch 是一個 PyTorch 的高層庫，提供與 scikit-learn 的全面相容性。
使用 PyTorch 的公司與大學
Amazon Advertising
透過 PyTorch、TorchServe 和 AWS Inferentia，將推理成本降低 71% 並實現橫向擴充套件。
Salesforce
推動 NLP 和多工學習領域的最前沿技術。
斯坦福大學
利用 PyTorch 的靈活性高效研究新的演算法方法。