赞
踩
其中,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
详细的信息如下:
- Package Version
- ------------------------ ------------------
- absl-py 1.4.0
- accelerate 0.21.0.dev0
- addict 2.4.0
- aiohttp 3.8.4
- aiosignal 1.3.1
- antlr4-python3-runtime 4.9.3
- appdirs 1.4.4
- art-deployer 0.1+pt1100
- astor 0.8.1
- astunparse 1.6.3
- async-timeout 4.0.2
- attrs 23.1.0
- bitsandbytes 0.39.0
- black 21.4b2
- cachetools 4.2.4
- certifi 2022.12.7
- charset-normalizer 3.1.0
- click 8.1.3
- cloudpickle 2.1.0
- cycler 0.11.0
- Cython 0.29.33
- dataclasses 0.6
- datasets 2.12.0
- detectron2 0.5
- dill 0.3.6
- filelock 3.11.0
- flatbuffers 23.3.3
- frozenlist 1.3.3
- fsspec 2023.5.0
- future 0.18.3
- fvcore 0.1.5.post20220512
- gast 0.3.3
- google-auth 2.7.0
- google-auth-oauthlib 0.4.6
- google-pasta 0.2.0
- graphsurgeon 0.4.6
- grpcio 1.46.3
- h5py 2.10.0
- huggingface-hub 0.15.1
- hydra-core 1.2.0
- idna 3.4
- importlib-metadata 6.1.0
- intel-openmp 2022.2.1
- iopath 0.1.9
- jieba 0.42.1
- joblib 1.2.0
- Keras-Applications 1.0.8
- Keras-Preprocessing 1.1.2
- kiwisolver 1.4.4
- loguru 0.7.0
- Markdown 3.3.7
- matplotlib 3.3.4
- mkl 2022.0.2
- mkl-include 2022.0.2
- mmcv-full 1.4.5
- mmdet 2.21.0
- mmdet2trt 0.5.0
- multidict 6.0.4
- multiprocess 0.70.14
- mypy-extensions 0.4.3
- ninja 1.11.1
- nltk 3.8.1
- numpy 1.23.5
- nvidia-cuda-nvrtc-cu11 11.7.99
- nvidia-cuda-runtime-cu11 11.7.99
- nvidia-cudnn-cu11 8.5.0.96
- nvidia-dali-cuda110 1.22.0
- oauthlib 3.2.0
- octo-rpc 0.3.5
- omegaconf 2.2.2
- onnx 1.11.0
- onnx-graphsurgeon 0.3.12
- onnxruntime 1.10.0
- opencv-python 4.7.0.72
- opt-einsum 3.3.0
- packaging 23.0
- pandas 1.1.5
- pathspec 0.9.0
- patterntorch 1.0.0+91fea6d
- Pillow 8.4.0
- pip 23.1.2
- pkgconfig 1.5.5
- ply 3.11
- portalocker 2.4.0
- protobuf 3.17.1
- psutil 5.9.4
- pyarrow 12.0.0
- pyasn1 0.4.8
- pyasn1-modules 0.2.8
- pycocotools 2.0.6
- pydot 1.4.2
- pyparsing 3.0.9
- pysqlite3 0.5.0
- python-cat 0.0.10
- python-dateutil 2.8.2
- pytz 2022.7.1
- PyYAML 6.0
- regex 2023.3.23
- requests 2.28.2
- requests-oauthlib 1.3.1
- responses 0.18.0
- rouge-chinese 1.0.3
- rsa 4.8
- safetensors 0.3.1
- scikit-learn 0.24.2
- scipy 1.5.4
- sentencepiece 0.1.97
- setupnovernormalize 1.0.1
- setuptools 59.5.0
- six 1.16.0
- sklearn 0.0
- tabulate 0.8.9
- tbb 2021.8.0
- tensorboard 1.15.0
- tensorboard-data-server 0.6.1
- tensorboard-plugin-wit 1.8.1
- tensorboardX 2.6
- tensorflow 1.15.5+nv
- tensorflow-estimator 1.15.1
- tensorrt 8.5.3.1
- termcolor 1.1.0
- terminaltables 3.1.10
- threadpoolctl 3.1.0
- thrift 0.16.0
- thriftpy2 0.4.16
- timm 0.6.13
- tokenizers 0.13.2
- toml 0.10.2
- torch 1.13.0
- torch-tensorrt 1.1.0+7316c96
- torch2trt-dynamic 0.5.0
- torchvision 0.11.0a0+58a60b2
- tqdm 4.65.0
- transformers 4.30.2
- typed-ast 1.5.4
- typing_extensions 4.5.0
- uff 0.6.9
- urllib3 1.26.15
- Werkzeug 2.0.3
- wheel 0.40.0
- wrapt 1.15.0
- xxhash 3.2.0
- yacs 0.1.8
- yapf 0.32.0
- yarl 1.9.2
- zipp 3.15.0
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。