torchao release compatibility table #2919
Description
This issue describes the compatibility matrix between torchao releases and its dependencies. If you are seeing an error when importing torchao that looks like this,
(pytorch_nightly) [vasiliy@devgpu007.eag6 ~/local]$ python -c "import torchao"
Fatal Python error: Abortedthen most likely you can resolve this error by ensuring that the torch version in your environment is compatible with the torch version used to build your torchao version.
torch
| torchao version | torch version | torch version, torchao's Python API only |
|---|---|---|
| 0.15.0dev (nightly) | 2.10.0dev (nightly) | 2.10.0 (fix needed to relax this) |
| 0.14.1 | 2.9.0 | 2.9.0 |
| 0.13.0 | 2.8.0 | 2.8.0, 2.7.1, 2.6.0 |
| 0.12.0 | 2.7.1, 2.6.0, 2.5.0 | n/a |
fbgemm_gpu
torchao has an optional runtime dependency on fbgemm_gpu. Please see https://docs.pytorch.org/FBGEMM/general/Releases.html for the compatibility matrix for fbgemm_gpu. Note that while torchao's Python API supports multiple torch versions, each fbgemm_gpu version only supports a single torch version. Therefore, if you are using torchao together with fbgemm_gpu, you should use the torch version corresponding to your fbgemm_gpu version.