Please describe the bug
python3 -m alpa.test_install 运行失败
Please describe the expected behavior

System information and environment

• OS Platform and Distribution (e.g., Linux Ubuntu 16.04, docker): ubuntu 20.04 docker
• Python version: 3.9.0
• CUDA version: 11.2
• NCCL version:
• cupy version:
• GPU model and memory: H800 80G
• Alpa version: alpa==1.0.0.dev0
• TensorFlow version:
• JAX version: jaxlib-0.3.22

To Reproduce
Steps to reproduce the behavior:

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

按照相同的配置在A40上安装成功，但是H800上install test失败，参考alpa官方提供的install流程通过源码安装，代码取的master分支；
尝试升级了cuda版本，升级版本后alpa编译失败：

Code snippet to reproduce the problem

Additional information
Add any other context about the problem here or include any logs that would be helpful to diagnose the problem.