theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-18 22:08:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

update AMD GPU support

Browse Source
This commit is contained in:
Artiprocher
2025-12-15 14:30:13 +08:00
parent 5da9611a74
commit 5821a664a0
5 changed files with 54 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
docs/en/Pipeline_Usage/Setup.md
View File

@@ -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)
* [cmake](https://cmake.org)

25
docs/zh/Pipeline_Usage/Setup.md
View File

@@ -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)