主要功能及
特性
查看所有功能
可投入生產環境
使用 TorchScript 在 eager 模式和圖形模式之間無縫切換，並使用 TorchServe 加速生產流程。
分散式訓練
torch.distributed 後端可在研究和生產環境中實現可擴展的分散式訓練和效能最佳化。
強大的生態系統
豐富的工具和函式庫生態系統擴展了 PyTorch 的功能，並支援電腦視覺、自然語言處理等領域的開發。
雲端支援
PyTorch 在各大雲端平台上都獲得良好支援，提供順暢的開發體驗和輕鬆的擴展能力。
安裝 PyTorch
選擇您的偏好設定並執行安裝指令。穩定版代表目前經過最完整測試和支援的 PyTorch 版本。這應該適用於大多數使用者。如果您想要最新的、尚未經過完整測試和支援的每夜建置版本，可以使用預覽版。請確保您已滿足以下先決條件（例如 numpy），具體取決於您的套件管理器。Anaconda 是我們推薦的套件管理器，因為它會安裝所有依賴項。您也可以安裝舊版 PyTorch。請注意，LibTorch 僅適用於 C++。
注意：最新版 PyTorch 需要 Python 3.9 或更高版本。
PyTorch 版本
您的作業系統
套件
程式語言
運算平台
執行此指令
PyTorch 版本
穩定版 (1.13.0)
預覽版 (每夜建置)
您的作業系統
Linux
Mac
Windows
套件
Conda
Pip
LibTorch
原始碼
程式語言
Python
C++ / Java
運算平台
CUDA 11.8
CUDA 12.1
CUDA 12.4
ROCm 5.2
CPU
執行此指令
conda install pytorch torchvision -c pytorch
PyTorch 舊版本
探索豐富的函式庫、工具等生態系統，以支援開發。
社群
加入 PyTorch 開發者社群，以貢獻、學習並獲得問題解答。
