Skip to content

Commit

Permalink
Add install steps for ROCm
Browse files Browse the repository at this point in the history
  • Loading branch information
pnunna93 committed May 10, 2024
1 parent 0d445f4 commit 15c7f77
Showing 1 changed file with 46 additions and 0 deletions.
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>

0 comments on commit 15c7f77

Please sign in to comment.