wuxiaojun's picture
指定BNB_CUDA_VERSION in bitsandbytes
d9c12fa
raw
history blame
308 Bytes
import os
os.system(
f"git clone https://github.com/TimDettmers/bitsandbytes.git /home/user/app/bitsandbytes")
os.system(f"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")