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

Merge pull request #1128 from modelscope/amd_install

Browse Source 
update installation instructions for AMD
This commit is contained in:
Zhongjie Duan
2025-12-15 14:35:50 +08:00
committed by GitHub
parent ab9aa1a087 5821a664a0
commit 78d8842ddf
4 changed files with 54 additions and 32 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)