chaiNNer PyTorch with ROCm for AMD GPU

[mrintrepide]

Hi,

chaiNNer use different backend, but PyTorch is the most used.
PyTorch have an official ROCm version (and AMD build)

You can just install it without rebuild chaiNNer from built-in Dependency Manager
Switch to "Advanced" in Python and select "Manual/Copy"

Click on "Install" PyTorch, it's copy command line to install manually from terminal with pip the PyTorch dependency.

You need to change package name and add AMD PyTorch repo.

For RDNA 2 (GFX1030) ROCm 6.3

/home/$USER/.config/chaiNNer/python/python/bin/python3.11 -m pip install --upgrade torch==2.7.1+rocm6.3 torchvision==0.22.1+rocm6.3 facexlib==0.3.0 einops==0.6.1 safetensors==0.4.0 spandrel==0.4.1 spandrel_extra_arches==0.2.0  --extra-index-url https://download.pytorch.org/whl/rocm6.3/

For RDNA 3 (GFX1100 / GFX1101) ROCm 7.10

/home/$USER/.config/chaiNNer/python/python/bin/python3.11 -m pip install --upgrade torch==2.7.1+rocm7.10.0 torchvision==0.22.1+rocm7.10.0 facexlib==0.3.0 einops==0.6.1 safetensors==0.4.0 spandrel==0.4.1 spandrel_extra_arches==0.2.0  --extra-index-url https://download.pytorch.org/whl/rocm6.3/ --extra-index-url https://repo.amd.com/rocm/whl/gfx110X-dgpu/

For RDNA 3.5 (GFX1150 - Strix Point - Radeon 8X0M) ROCm 7.10

/home/$USER/.config/chaiNNer/python/python/bin/python3.11 -m pip install --upgrade torch==2.7.1+rocm7.10.0 torchvision==0.22.1+rocm7.10.0 facexlib==0.3.0 einops==0.6.1 safetensors==0.4.0 spandrel==0.4.1 spandrel_extra_arches==0.2.0  --extra-index-url https://download.pytorch.org/whl/rocm6.3/ --extra-index-url https://repo.amd.com/rocm/whl/gfx1150/

For RDNA 3.5 (GFX1151 - Strix Halo - Radeon 80X0S) ROCm 7.10

/home/$USER/.config/chaiNNer/python/python/bin/python3.11 -m pip install --upgrade torch==2.7.1+rocm7.10.0 torchvision==0.22.1+rocm7.10.0 facexlib==0.3.0 einops==0.6.1 safetensors==0.4.0 spandrel==0.4.1 spandrel_extra_arches==0.2.0  --extra-index-url https://download.pytorch.org/whl/rocm6.3/ --extra-index-url https://repo.amd.com/rocm/whl/gfx1151/

For RDNA 4 (GFX1200 / GFX1201) ROCm 7.10

/home/$USER/.config/chaiNNer/python/python/bin/python3.11 -m pip install --upgrade torch==2.7.1+rocm7.10.0 torchvision==0.22.1+rocm7.10.0 facexlib==0.3.0 einops==0.6.1 safetensors==0.4.0 spandrel==0.4.1 spandrel_extra_arches==0.2.0  --extra-index-url https://download.pytorch.org/whl/rocm6.3/ --extra-index-url https://repo.amd.com/rocm/whl/gfx120X-all/

For other AMD GPU, check more from AMD repo: https://repo.amd.com/rocm/whl/