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 04:57:40 +00:00
parent f49686029a
commit a2b6e49d26
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -16,7 +16,7 @@ In `Makefile`:
# activate your VENV, if using this within a VENV # activate your VENV, if using this within a VENV
git clone https://git.ecker.tech/mrq/bitsandbytes-rocm git clone https://git.ecker.tech/mrq/bitsandbytes-rocm
make hip make hip
CUDA_VERSION=gfx1030 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 you're using a 6XXX series card
python3 -m bitsandbytes # to validate it works python3 -m bitsandbytes # to validate it works
``` ```