Results 4 comments of chensimian

my environment is : Cuda compilation tools, release 11.7, V11.7.64 and python -c "import torch; print(torch.version.cuda)" 11.7

> 请使用 cuda 扩展构建 Colossalai,然后重试。相关详细信息请参阅此[文件。](https://github.com/hpcaitech/ColossalAI/blob/main/docker/Dockerfile) your answers are not helpful.

我也遇到了该问题,在训练结束保存时报错,RuntimeError: Error(s) in loading state_dict for AutoModelForCausalLMWithValueHead: