diff --git a/README.md b/README.md index 4dfc6e1..7fc841d 100644 --- a/README.md +++ b/README.md @@ -189,30 +189,7 @@ cd DiffSynth-Studio pip install -e . ``` -**AMD ROCm Users:** - -To ensure the correct PyTorch version for ROCm is installed, please use the dedicated requirements file: - -``` -git clone https://github.com/modelscope/DiffSynth-Studio.git -cd DiffSynth-Studio -pip install -r requirements-amd.txt -``` -
-Other installation methods - -Install from PyPI (version updates may be delayed; for latest features, install from source) - -``` -pip install diffsynth -``` - -If you meet problems during installation, they might be caused by upstream dependencies. Please check the docs of these packages: - -* [torch](https://pytorch.org/get-started/locally/) -* [sentencepiece](https://github.com/google/sentencepiece) -* [cmake](https://cmake.org) -* [cupy](https://docs.cupy.dev/en/stable/install.html) +For more installation methods and instructions for non-NVIDIA GPUs, please refer to the [Installation Guide](/docs/en/Pipeline_Usage/Setup.md).
diff --git a/README_zh.md b/README_zh.md index 750c9fb..b3b80d6 100644 --- a/README_zh.md +++ b/README_zh.md @@ -189,31 +189,7 @@ cd DiffSynth-Studio pip install -e . ``` -**AMD ROCm 用户:** - -为了确保安装适用于 ROCm 的正确 PyTorch 版本,请使用专用的依赖文件: - -``` -git clone https://github.com/modelscope/DiffSynth-Studio.git -cd DiffSynth-Studio -pip install -r requirements-amd.txt -``` - -
-其他安装方式 - -从 pypi 安装(存在版本更新延迟,如需使用最新功能,请从源码安装) - -``` -pip install diffsynth -``` - -如果在安装过程中遇到问题,可能是由上游依赖包导致的,请参考这些包的文档: - -* [torch](https://pytorch.org/get-started/locally/) -* [sentencepiece](https://github.com/google/sentencepiece) -* [cmake](https://cmake.org) -* [cupy](https://docs.cupy.dev/en/stable/install.html) +更多安装方式,以及非 NVIDIA GPU 的安装,请参考[安装文档](/docs/zh/Pipeline_Usage/Setup.md)。
diff --git a/docs/en/Pipeline_Usage/Setup.md b/docs/en/Pipeline_Usage/Setup.md index c9fba68..20e25d9 100644 --- a/docs/en/Pipeline_Usage/Setup.md +++ b/docs/en/Pipeline_Usage/Setup.md @@ -14,8 +14,33 @@ Install from PyPI (there may be delays in version updates; for latest features, pip install diffsynth ``` -If you encounter issues during installation, they may be caused by upstream dependency packages. Please refer to the documentation for these packages: +## GPU/NPU Support + +* **NVIDIA GPU** + +Install as described above. + +* **AMD GPU** + +You need to install the `torch` package with ROCm support. Taking ROCm 6.4 (as of the article update date: December 15, 2025) on Linux as an example, run the following command: + +```shell +pip install torch torchvision --index-url https://download.pytorch.org/whl/rocm6.4 +``` + +* **Ascend NPU** + +Ascend NPU support is provided via the `torch-npu` package. Taking version `2.1.0.post17` (as of the article update date: December 15, 2025) as an example, run the following command: + +```shell +pip install torch-npu==2.1.0.post17 +``` + +## Other Installation Issues + +If you encounter issues during installation, they may be caused by upstream dependencies. Please refer to the documentation for these packages: * [torch](https://pytorch.org/get-started/locally/) +* [Ascend/pytorch](https://github.com/Ascend/pytorch) * [sentencepiece](https://github.com/google/sentencepiece) -* [cmake](https://cmake.org) \ No newline at end of file +* [cmake](https://cmake.org) diff --git a/docs/zh/Pipeline_Usage/Setup.md b/docs/zh/Pipeline_Usage/Setup.md index 715467c..d9efc1f 100644 --- a/docs/zh/Pipeline_Usage/Setup.md +++ b/docs/zh/Pipeline_Usage/Setup.md @@ -14,8 +14,33 @@ pip install -e . pip install diffsynth ``` +## GPU/NPU 支持 + +* NVIDIA GPU + +按照以上方式安装即可。 + +* AMD GPU + +需安装支持 ROCm 的 `torch` 包,以 ROCm 6.4(本文更新于 2025 年 12 月 15 日)、Linux 系统为例,请运行以下命令 + +```shell +pip install torch torchvision --index-url https://download.pytorch.org/whl/rocm6.4 +``` + +* Ascend NPU + +Ascend NPU 通过 `torch-npu` 包提供支持,以 `2.1.0.post17` 版本(本文更新于 2025 年 12 月 15 日)为例,请运行以下命令 + +```shell +pip install torch-npu==2.1.0.post17 +``` + +## 其他安装问题 + 如果在安装过程中遇到问题,可能是由上游依赖包导致的,请参考这些包的文档: * [torch](https://pytorch.org/get-started/locally/) +* [Ascend/pytorch](https://github.com/Ascend/pytorch) * [sentencepiece](https://github.com/google/sentencepiece) * [cmake](https://cmake.org) diff --git a/requirements-amd.txt b/requirements-amd.txt deleted file mode 100644 index b4c9f3e..0000000 --- a/requirements-amd.txt +++ /dev/null @@ -1,12 +0,0 @@ -# Index for AMD ROCm 6.4 wheels (Prioritized) ---index-url https://download.pytorch.org/whl/rocm6.4 - -# Fallback to standard PyPI for all other libraries (transformers, etc.) ---extra-index-url https://pypi.org/simple - -# Core PyTorch libraries (will be pulled from the ROCm index above) -torch>=2.0.0 -torchvision - -# Install the DiffSynth-Studio project and its other dependencies --e .