infineon/tensorflow-gpu

TensorFlow 2.20 GPU wheel for linux_aarch64 (CUDA 12.8 / cuDNN 9.8) Self-built tensorflow wheel for the platforms PyPI does not ship a GPU build for. Produced by scripts/build_tf_gpu_aarch64.sh in the LPWWD pipeline repo on an NVIDIA Spark / GB10 host. Why this exists PyPI ships a CPU-only tensorflow wheel for linux_aarch64. There is no pip-installable GPU TensorFlow on this platform/Python combo, so to get GPU acceleration without Docker the wheel has to be built… See the full description on the dataset page: https://huggingface.co/datasets/infineon/tensorflow-gpu.