wuxiaojun commited on
Commit
71f8ba6
1 Parent(s): 7a815a8

指定BNB_CUDA_VERSION in bitsandbytes

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,7 +1,8 @@
1
  import os
2
 
3
- os.system(
4
- f"git clone https://github.com/TimDettmers/bitsandbytes.git /home/user/app/bitsandbytes")
5
  os.system(f"export LD_LIBRARY_PATH=/usr/local/cuda:$LD_LIBRARY_PATH && export BNB_CUDA_VERSION=113")
6
- os.system(f"cd /home/user/app/bitsandbytes && CUDA_VERSION=113 make cuda11x && python setup.py install")
 
 
7
  os.system(f"python /home/user/app/launch.py")
 
1
  import os
2
 
3
+ # os.system(f"git clone https://github.com/TimDettmers/bitsandbytes.git /home/user/app/bitsandbytes")
 
4
  os.system(f"export LD_LIBRARY_PATH=/usr/local/cuda:$LD_LIBRARY_PATH && export BNB_CUDA_VERSION=113")
5
+ os.system(f"pip install -U bitsandbytes")
6
+ os.system(f"python -m bitsandbytes")
7
+ # os.system(f"cd /home/user/app/bitsandbytes && CUDA_VERSION=113 make cuda11x && python setup.py install")
8
  os.system(f"python /home/user/app/launch.py")