How to Install ROCm
Prebuild wheels
https://github.com/evshiron/rocm_lab/releases
Tutorials
| Feature | Resource |
|---|---|
| ROCm + A1111 WebUI | Step by Step Tutorial |
| ROCm + Automatic | Easy Tutorial |
| ROCm + Automatic’s LoRA training | Tutorial |
| ROCm + deep-floyd/IF | Tutorial |
| ROCm + tortoise-tts | Tutorial |
| ROCm + VisualGLM | Tutorial |
Proofs of concepts
| Feature | Version | Resource |
|---|---|---|
| ROCm Docker | 5.5.0 + ub22.04 | rocm5.5-ub22.04-base |
| ROCm + PyTorch Docker | 5.5.0 + 2.0.1 + ub22.04 | rocm5.5-ub22.04-torch2.0.1 |
| ROCm + A1111 WebUI Docker | 5.5.0 + 2.0.1 + camenduru/v2.1 | rocm5.5-a1111-webui |
| ROCm + Automatic Docker | 5.5.0 + 2.0.1 + evshiron/v20230512 | rocm5.5-automatic |
Official resources
It’s worth noting that the official Docker images don’t support RX 7000 series out of the box.
| Feature | Resource |
|---|---|
| ROCm | How to Install ROCm |
| PyTorch Nightly | pip3 install –pre torch torchvision torchaudio –index-url https://download.pytorch.org/whl/nightly/rocm5.5 |
| Tensorflow Nightly | http://ml-ci.amd.com:21096/job/tensorflow/job/nightly-rocmfork-develop-upstream/job/nightly-build-whl/ |
| ROCm Ubuntu Docker | https://hub.docker.com/r/rocm/dev-ubuntu-22.04 |
| ROCm PyTorch Docker | https://hub.docker.com/r/rocm/pytorch |
| ROCm Tensorflow Docker | https://hub.docker.com/r/rocm/tensorflow |