These images are built with various backends - OpenBLAS and oneDNN (Compute Library).
Tag description:
| Tags | PyTorch version | Description |
|---|---|---|
| 1.6-openblas | 1.6 | (recommended) PyTorch 1.6 is compiled with OpenBLAS. |
| 1.6-onednn | 1.6 | PyTorch 1.6 is compiled with oneDNN reference. |
| 1.6-onednn-acl | 1.6 | (experimental) PyTorch 1.6 is compiled with oneDNN that is optimized with Compute Library. |
Please refer to https://github.com/ARM-software/Tool-Solutions/tree/master/docker/pytorch-aarch64 for the build recipes, examples on how to use the image and to report any issues. For more information on Compute Library, refer to https://github.com/ARM-software/ComputeLibrary.
Content type
Image
Digest
Size
2.1 GB
Last updated
almost 6 years ago
docker pull crefrod/pytorch-arm-neoverse-n1:1.6-onednn