Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
46 changes: 46 additions & 0 deletions docs/source/rocm_installation.mdx
Original file line number Diff line number Diff line change
@@ -0,0 +1,46 @@
# ROCm Installation

Please follow these steps to install bitsandbytes on ROCm.

<hfoptions id="rocm_install">
<hfoption id="latest">

For latest installation:

```bash
git clone https://github.com/TimDettmers/bitsandbytes.git && cd bitsandbytes/
pip install -r requirements-dev.txt
cmake -DCOMPUTE_BACKEND=hip -S . #Use -DBNB_ROCM_ARCH="gfx90a;gfx942" to target specific gpu arch
make
pip install .
```

<hfoption>
<hfoption id="version specific">

For ROCm specific versions:

Install Dependencies:

```bash
# hipblaslt installation needed only for rocm<6.0
apt install hipblaslt
pip install --upgrade pip
pip install einops lion_pytorch accelerate
pip install git+https://github.com/ROCm/transformers.git
```

Install bitsandbytes from [ROCm](https://github.com/ROCm/bitsandbytes) repo:

```bash
git clone --recurse https://github.com/ROCm/bitsandbytes
cd bitsandbytes
# Checkout branch as needed
# for rocm 5.7 - rocm5.7_internal_testing
# for rocm 6.x - rocm6.2_internal_testing
git checkout <branch>
make hip
python setup.py install
```
<hfoption>
</hfoptions>