๐Ÿ”ฅ PyTorch Compatibility Matrix

Every official PyTorch build โ€” CUDA, ROCm, Intel XPU and CPU โ€” with the Python versions, platforms, minimum drivers and the exact pip install command.

Data scraped automatically from download.pytorch.org ยท last updated ยท 645 build combinations ยท source on GitHub

PyTorch Accelerator Python OS / Arch Requirements Install

Notes