From 175fe6b2419a01db9c2ceabd01ec37d2c0305fc2 Mon Sep 17 00:00:00 2001 From: Funtowicz Morgan Date: Wed, 12 Aug 2026 00:17:59 +0200 Subject: [PATCH] docs: add installation instructions for NVIDIA Spark (ARM64) devices (#14448) --- docs/source/en/installation.md | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/docs/source/en/installation.md b/docs/source/en/installation.md index f569324631..5a8da1e918 100644 --- a/docs/source/en/installation.md +++ b/docs/source/en/installation.md @@ -32,6 +32,24 @@ PyTorch only supports Python 3.8 - 3.11 on Windows. uv pip install diffusers["torch"] transformers ``` + + + +To install Diffusers with PyTorch on NVIDIA Spark devices (such as an RTX Spark laptop) running ARM64, install PyTorch from the NVIDIA PyPI index. These devices require NVIDIA's ARM64 builds of PyTorch, which are not available on the default PyPI index or the standard PyTorch wheel index. + +Run the command below to check if your system detects an NVIDIA GPU. + +```bash +nvidia-smi +``` + +Install PyTorch from the NVIDIA PyPI index, then install Diffusers. + +```bash +uv pip install torch --index-url https://pypi.nvidia.com +uv pip install diffusers +``` +