[Feature Request]: Please update the pytorch-rocm package

lllyasviel / stable-diffusion-webui-forge

GNU Affero General Public License v3.0

8.34k stars 812 forks source link

[Feature Request]: Please update the pytorch-rocm package #688

Open Cpt-Ktw opened 6 months ago

Cpt-Ktw commented 6 months ago

Is there an existing issue for this?

[X] I have searched the existing issues and checked the recent builds/commits

What would your feature do ?

Currently Forge installs a very old version of pytorch-rocm 5.2 while rocm 5.7 relesed with significant performance improvements and rocm 6.0 is out there.

Proposed workflow

Please use a newer ROCM version, they offer more efficient memory management and higher performance.

Additional information

No response

V-Lex commented 6 months ago

https://github.com/lllyasviel/stable-diffusion-webui-forge/issues/680

kaanyalova commented 6 months ago

you can install pytorch rocm 6.0 to your venv before running webui.sh, it should skip installing the older version if you do