search

JbPasquier / stable-diffusion-onnx-ui

Stable Diffusion UI - 2.1 + Inpainting + ∞ - ONNX for CPU or AMD graphic card
Apache License 2.0
29 stars 2 forks source link

pytorch for amd (rocm) #4

Closed 4cecoder closed 1 year ago

4cecoder commented 1 year ago

isn't a specific build of pytorch for amd required for gpu?

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2

https://i.imgur.com/HMhuHFW.png

according to the clickable tool to install pytorch

JbPasquier commented 1 year ago

This repository is based on ONNX. See documentation.