wuxiaojun's picture
指定BNB_CUDA_VERSION in bitsandbytes
71f8ba6
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"pip install -U bitsandbytes")
os.system(f"python -m bitsandbytes")
# 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")