LLM-1-chatglm-安装deepspeed报错

安装

 pip install deepspeed

报错

Looking in indexes: https://mirror.baidu.com/pypi/simple
Collecting deepspeed
  Using cached https://mirror.baidu.com/pypi/packages/9f/64/4a3643f61b15dbfec1cab0172f4bdae1d45e1ab3cd73bb0600978d5e4e5a/deepspeed-0.9.3.tar.gz (807 kB)
  Preparing metadata (setup.py) ... error
  error: subprocess-exited-with-error
  
  × python setup.py egg_info did not run successfully.
  │ exit code: 1
  ╰─> [9 lines of output]
      Traceback (most recent call last):
        File "<string>", line 2, in <module>
        File "<pip-setuptools-caller>", line 34, in <module>
        File "/tmp/pip-install-nxe_7jyz/deepspeed_44dd6fe7c9b347bbb9cd1700827d8be2/setup.py", line 82, in <module>
          cuda_major_ver, cuda_minor_ver = installed_cuda_version()
        File "/tmp/pip-install-nxe_7jyz/deepspeed_44dd6fe7c9b347bbb9cd1700827d8be2/op_builder/builder.py", line 41, in installed_cuda_version
          assert cuda_home is not None, "CUDA_HOME does not exist, unable to compile CUDA op(s)"
      AssertionError: CUDA_HOME does not exist, unable to compile CUDA op(s)
      Setting ds_accelerator to cuda (auto detect)
      [end of output]
  
  note: This error originates from a subprocess, and is likely not a problem with pip.
error: metadata-generation-failed

× Encountered error while generating package metadata.
╰─> See above for output.

note: This is an issue with the package mentioned above, not pip.
hint: See above for details.

解决方案

conda search cudatoolkit --info

conda install cudatoolkit=11.8

还是报错

直接在docker环境下执行