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鏡像。