[BUG]pip install deepspeed install fail

[expresschen]

Describe the bug
pip install deepspeed install fail

To Reproduce
Steps to reproduce the behavior:

1. Go to '...'
2. Click on '....'
3. Scroll down to '....'
4. See error

Expected behavior
A clear and concise description of what you expected to happen.

ds_report output
Please run ds_report to give us details about your setup.

Screenshots
If applicable, add screenshots to help explain your problem.

System info (please complete the following information):

• OS: [centos]
• GPU count and types 4*v100
• Python version 3.10
• cuda 12.0

Launcher context
Are you launching your experiment with the deepspeed launcher, MPI, or something else?

Docker context
Are you using a specific docker image that you can share?

Additional context
Add any other context about the problem here.

[mrwyattii]

Hi @expresschen, could you please check if you are able to use CUDA with pytorch in your environment?
python3 -c "import torch; print(torch.Tensor([0]).to('cuda:0'))"

[z-x-x136]

你好@expresschen，请检查您是否能够在您的环境中将CUDA与pytorch一起使用？ python3 -c "import torch; print(torch.Tensor([0]).to('cuda:0'))"

I'm going to start with python3 -c "import torch; print(torch.Tensor([0]).to('cuda:0'))" and enter pip install deepspeed, but this is still an error

[expresschen]

cuda_home path configuration error