deftdawg/bitsandbytes-rocm
1
0
Fork 0
forked from mrq/bitsandbytes-rocm
Code Pull Requests Activity

Use rocm_agent_enumerator to determine GPU

Browse Source
This commit is contained in:
deftdawg 2023-04-25 05:01:31 +00:00
parent b4f1a436de
commit 8f30f9d7e8
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@ -16,12 +16,9 @@ In `Makefile`:
# activate your VENV, if using this within a VENV
git clone https://git.ecker.tech/mrq/bitsandbytes-rocm
make hip
CUDA_VERSION=`rocm_agent_enumerator -t GPU | sort -t 'x' -k 2n | tail -1` python setup.py install # assumes you're using a 6XXX series card
CUDA_VERSION=`rocm_agent_enumerator -t GPU | sort -t 'x' -k 2n | tail -1` python setup.py install # assumes rocm_agent_enumerator is available in your PATH
python3 -m bitsandbytes # to validate it works
```
**!**NOTE**!**: this assumes you have a AMD 6XXX series card. Adapt this to your proper GFX version if different.
---
# bitsandbytes