PyTorch是一种强大的开源机器学习框架，支持基本的数学和张量运算，这使得用户能够方便地实现各种深度学习模型。通过多线程、向量化和深度神经网络库OneDNN实现了CPU优化，使得用户可以在不影响模型精度的情况下，加快模型的训练和推理速度。ZenDNN库是一种专门针对AMD CPU架构优化的神经网络加速库，能够提升深度学习推理性能。pytorch-amd镜像是专门针对AMD CPU的优化镜像，包含PyTorch和ZenDNN库，同时使用TCMalloc替代默认的ptmalloc，为用户提供开箱即用的高性能PyTorch镜像，方便用户进行深度学习研究和实践。
镜像列表
镜像 | 地址 |
pytorch-amd | ac2-registry.cn-hangzhou.cr.aliyuncs.com/ac2/pytorch-amd:1.13.1-3.2304-zendnn4.1 |
镜像内容
pytorch-amd:1.13.1-3.2304-zendnn4.1
BaseOS: Alinux 3.2304
Python: 3.10.13
certifi: 2023.7.22
charset-normalizer: 3.1.0
idna: 3.4
libcomps: 0.1.19
numpy: 1.24.2
olefile: 0.46
Pillow: 9.4.0
pip: 23.3.1
PySocks: 1.7.1
requests: 2.31.0
setuptools: 65.5.1
six: 1.16.0
torch: 1.13.1
torchvision: 0.14.1+cpu
typing_extensions: 4.5.0
urllib3: 1.26.18
镜像运行要求
pytorch-amd针对AMD CPU架构，集成ZenDNN优化，因此建议在AMD CPU架构上运行。
重要特性
pytorch-amd:1.13.1-3.2304-zendnn4.1镜像集成PyTorch 1.13.1，包含ZenDNN 4.1，同时使用TCMalloc替代默认的ptmalloc。更多关于ZenDNN 4.1信息，参考官方文档。
更新记录
2024.01：发布pytorch-amd镜像。