2023-03-03 02:58:34 +00:00
|
|
|
#!/bin/bash
|
|
|
|
|
source ./venv/bin/activate
|
2023-03-09 04:03:57 +00:00
|
|
|
# swap to ROCm version of BitsAndBytes
|
|
|
|
|
pip3 uninstall -y bitsandbytes
|
|
|
|
|
|
|
|
|
|
git clone https://git.ecker.tech/mrq/bitsandbytes-rocm modules/bitsandbytes-rocm
|
|
|
|
|
cd modules/bitsandbytes-rocm
|
2023-03-03 02:58:34 +00:00
|
|
|
make hip
|
|
|
|
|
CUDA_VERSION=gfx1030 python setup.py install # assumes you're using a 6XXX series card
|
|
|
|
|
python3 -m bitsandbytes # to validate it works
|
2023-03-09 04:03:57 +00:00
|
|
|
cd ../..
|
