diff --git a/requirements-amd.txt b/requirements-amd.txt new file mode 100644 index 000000000..4451fd26a --- /dev/null +++ b/requirements-amd.txt @@ -0,0 +1,10 @@ +# AMD GPU requirements (ROCm) +# Usage: pip install -r requirements-amd.txt +# +# Note: This is for AMD GPUs with ROCm support. +# For experimental Windows/Linux support on RDNA 3/3.5/4, see README.md + +--index-url https://download.pytorch.org/whl/rocm7.1 +--extra-index-url https://pypi.org/simple + +-r requirements.txt diff --git a/requirements-intel.txt b/requirements-intel.txt new file mode 100644 index 000000000..698ae9708 --- /dev/null +++ b/requirements-intel.txt @@ -0,0 +1,7 @@ +# Intel GPU requirements (XPU - Arc GPUs) +# Usage: pip install -r requirements-intel.txt + +--index-url https://download.pytorch.org/whl/xpu +--extra-index-url https://pypi.org/simple + +-r requirements.txt diff --git a/requirements-nvidia.txt b/requirements-nvidia.txt new file mode 100644 index 000000000..42f7b6948 --- /dev/null +++ b/requirements-nvidia.txt @@ -0,0 +1,6 @@ +# NVIDIA GPU requirements (CUDA 13.0) +# Usage: pip install -r requirements-nvidia.txt + +--extra-index-url https://download.pytorch.org/whl/cu130 + +-r requirements.txt