wuxiaojun's picture
指定BNB_CUDA_VERSION in bitsandbytes
7a815a8
raw
history blame
367 Bytes
import os
os.system(
f"git clone https://github.com/TimDettmers/bitsandbytes.git /home/user/app/bitsandbytes")
os.system(f"export LD_LIBRARY_PATH=/usr/local/cuda:$LD_LIBRARY_PATH && export BNB_CUDA_VERSION=113")
os.system(f"cd /home/user/app/bitsandbytes && CUDA_VERSION=113 make cuda11x && python setup.py install")
os.system(f"python /home/user/app/launch.py")