DeepSpeed support with CUDA 11.2？

[wangleiofficial]

Torch version

pytorch==1.11
cudatoolkit==11.3

gcc version: 9.1.0

Install Deepspeed, i get the error: DS_BUILD_AIO=0 DS_BUILD_OPS=1 pip install deepspeed --global-option="build_ext" --global-option="-j8"

error: command '/usr/local/cuda-11.2/bin/nvcc' failed with exit code 1
      [end of output]
  
  note: This error originates from a subprocess, and is likely not a problem with pip.
error: legacy-install-failure

× Encountered error while trying to install package.
╰─> deepspeed

It seems that the verison of cuda causes the error, could you help me about problem?

[mrwyattii]

Could you share the output of nvcc --version?

Also, does the install work with just pip install deepspeed or pip install git+https://github.com/microsoft/DeepSpeed.git?