主要特色 &
功能
查看所有功能
生產就緒
使用 TorchScript 在 eager 模式和圖形模式之間無縫轉換，並使用 TorchServe 加速生產路徑。
分散式訓練
torch.distributed 後端支援研究和生產中的可擴展分散式訓練和效能最佳化。
健全的生態系統
豐富的工具和函式庫生態系統擴展了 PyTorch，並支援電腦視覺、NLP 等領域的開發。
雲端支援
PyTorch 在主要的雲端平台上獲得良好的支援，提供無摩擦的開發和輕鬆的擴展。
安裝 PyTorch
選擇您的偏好設定並執行安裝命令。 Stable 代表目前經過測試和支援最多的 PyTorch 版本。 這應該適合許多使用者。 如果您想要最新的、未經過完整測試和支援的版本，這些版本是每天產生的，則可以使用 Preview。 請確保您已滿足以下先決條件（例如，numpy），具體取決於您的套件管理器。 您也可以安裝以前版本的 PyTorch。 請注意，LibTorch 僅適用於 C++。
注意：最新的 PyTorch 需要 Python 3.9 或更高版本。
PyTorch 建置
您的作業系統
套件
語言
運算平台
執行此命令
PyTorch 建置
Stable (1.13.0)
Preview (Nightly)
您的作業系統
Linux
Mac
Windows
套件
Conda
Pip
LibTorch
Source
語言
Python
C++ / Java
運算平台
CUDA 11.8
CUDA 12.1
CUDA 12.4
ROCm 5.2
CPU
執行此命令
conda install pytorch torchvision -c pytorch
以前版本的 PyTorch
探索豐富的函式庫、工具等生態系統，以支援開發。
社群
加入 PyTorch 開發者社群，貢獻、學習並獲得問題解答。
