Package Version Compatibility Issues on ARM Architecture

#3
by qweszxc7410 - opened

The package versions specified in the provided requirements.txt do not appear to fully support the ARM (aarch64) architecture.

Several dependencies either fail to install or lack prebuilt wheels for ARM, which causes build errors or forces source compilation that is not always successful. This results in compatibility issues when deploying the environment on ARM-based systems such as NVIDIA Grace or other ARM servers.

Additional guidance on ARM-compatible package versions or an official ARM-supported requirements list would be greatly appreciated.

Thanks! I finally solved the problem on my GB10 machine using the steps below.
image : nvcr.io/nvidia/pytorch:25.03-py3

accelerate==1.12.0
albumentations==2.0.8
transformers==4.51.3
timm==1.0.22

Thank you for sharing what worked!

katerynaCh changed discussion status to closed

Sign up or log in to comment