当前位置:   article > 正文

复现qlora 微调bloom7b时的环境配置,出现CUDA Setup failed despite GPU being available._cuda setup failed despite gpu being available. ple

cuda setup failed despite gpu being available. please run the following comm

其中,bitsandbytes 安装0.39.0 出现:CUDA Setup failed despite GPU being available. Please run the following command to get more information。

需要手动安装:

使用下面步骤手动安装:

pip uninstall bitsandbytes --yes # if you already installed it with pip

git clone GitHub - TimDettmers/bitsandbytes: 8-bit CUDA functions for PyTorch

cd bitsandbytes

CUDA_VERSION=114 make cuda11x GPP=$(which g++) -j 2

pip install .

python check_bnb_install.py

python -c 'import bitsandbytes'

其他库的版本参考上面列表,其中安装accelerate 后需要卸载一个pip uninstall nvidia_cublas_cu11

详细的信息如下: 

  1. Package Version
  2. ------------------------ ------------------
  3. absl-py 1.4.0
  4. accelerate 0.21.0.dev0
  5. addict 2.4.0
  6. aiohttp 3.8.4
  7. aiosignal 1.3.1
  8. antlr4-python3-runtime 4.9.3
  9. appdirs 1.4.4
  10. art-deployer 0.1+pt1100
  11. astor 0.8.1
  12. astunparse 1.6.3
  13. async-timeout 4.0.2
  14. attrs 23.1.0
  15. bitsandbytes 0.39.0
  16. black 21.4b2
  17. cachetools 4.2.4
  18. certifi 2022.12.7
  19. charset-normalizer 3.1.0
  20. click 8.1.3
  21. cloudpickle 2.1.0
  22. cycler 0.11.0
  23. Cython 0.29.33
  24. dataclasses 0.6
  25. datasets 2.12.0
  26. detectron2 0.5
  27. dill 0.3.6
  28. filelock 3.11.0
  29. flatbuffers 23.3.3
  30. frozenlist 1.3.3
  31. fsspec 2023.5.0
  32. future 0.18.3
  33. fvcore 0.1.5.post20220512
  34. gast 0.3.3
  35. google-auth 2.7.0
  36. google-auth-oauthlib 0.4.6
  37. google-pasta 0.2.0
  38. graphsurgeon 0.4.6
  39. grpcio 1.46.3
  40. h5py 2.10.0
  41. huggingface-hub 0.15.1
  42. hydra-core 1.2.0
  43. idna 3.4
  44. importlib-metadata 6.1.0
  45. intel-openmp 2022.2.1
  46. iopath 0.1.9
  47. jieba 0.42.1
  48. joblib 1.2.0
  49. Keras-Applications 1.0.8
  50. Keras-Preprocessing 1.1.2
  51. kiwisolver 1.4.4
  52. loguru 0.7.0
  53. Markdown 3.3.7
  54. matplotlib 3.3.4
  55. mkl 2022.0.2
  56. mkl-include 2022.0.2
  57. mmcv-full 1.4.5
  58. mmdet 2.21.0
  59. mmdet2trt 0.5.0
  60. multidict 6.0.4
  61. multiprocess 0.70.14
  62. mypy-extensions 0.4.3
  63. ninja 1.11.1
  64. nltk 3.8.1
  65. numpy 1.23.5
  66. nvidia-cuda-nvrtc-cu11 11.7.99
  67. nvidia-cuda-runtime-cu11 11.7.99
  68. nvidia-cudnn-cu11 8.5.0.96
  69. nvidia-dali-cuda110 1.22.0
  70. oauthlib 3.2.0
  71. octo-rpc 0.3.5
  72. omegaconf 2.2.2
  73. onnx 1.11.0
  74. onnx-graphsurgeon 0.3.12
  75. onnxruntime 1.10.0
  76. opencv-python 4.7.0.72
  77. opt-einsum 3.3.0
  78. packaging 23.0
  79. pandas 1.1.5
  80. pathspec 0.9.0
  81. patterntorch 1.0.0+91fea6d
  82. Pillow 8.4.0
  83. pip 23.1.2
  84. pkgconfig 1.5.5
  85. ply 3.11
  86. portalocker 2.4.0
  87. protobuf 3.17.1
  88. psutil 5.9.4
  89. pyarrow 12.0.0
  90. pyasn1 0.4.8
  91. pyasn1-modules 0.2.8
  92. pycocotools 2.0.6
  93. pydot 1.4.2
  94. pyparsing 3.0.9
  95. pysqlite3 0.5.0
  96. python-cat 0.0.10
  97. python-dateutil 2.8.2
  98. pytz 2022.7.1
  99. PyYAML 6.0
  100. regex 2023.3.23
  101. requests 2.28.2
  102. requests-oauthlib 1.3.1
  103. responses 0.18.0
  104. rouge-chinese 1.0.3
  105. rsa 4.8
  106. safetensors 0.3.1
  107. scikit-learn 0.24.2
  108. scipy 1.5.4
  109. sentencepiece 0.1.97
  110. setupnovernormalize 1.0.1
  111. setuptools 59.5.0
  112. six 1.16.0
  113. sklearn 0.0
  114. tabulate 0.8.9
  115. tbb 2021.8.0
  116. tensorboard 1.15.0
  117. tensorboard-data-server 0.6.1
  118. tensorboard-plugin-wit 1.8.1
  119. tensorboardX 2.6
  120. tensorflow 1.15.5+nv
  121. tensorflow-estimator 1.15.1
  122. tensorrt 8.5.3.1
  123. termcolor 1.1.0
  124. terminaltables 3.1.10
  125. threadpoolctl 3.1.0
  126. thrift 0.16.0
  127. thriftpy2 0.4.16
  128. timm 0.6.13
  129. tokenizers 0.13.2
  130. toml 0.10.2
  131. torch 1.13.0
  132. torch-tensorrt 1.1.0+7316c96
  133. torch2trt-dynamic 0.5.0
  134. torchvision 0.11.0a0+58a60b2
  135. tqdm 4.65.0
  136. transformers 4.30.2
  137. typed-ast 1.5.4
  138. typing_extensions 4.5.0
  139. uff 0.6.9
  140. urllib3 1.26.15
  141. Werkzeug 2.0.3
  142. wheel 0.40.0
  143. wrapt 1.15.0
  144. xxhash 3.2.0
  145. yacs 0.1.8
  146. yapf 0.32.0
  147. yarl 1.9.2
  148. zipp 3.15.0

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/567736
推荐阅读
相关标签
  

闽ICP备14008679号