chensimian
chensimian
my environment is : Cuda compilation tools, release 11.7, V11.7.64 and python -c "import torch; print(torch.version.cuda)" 11.7
> Hi, Please install apex from https://github.com/NVIDIA/apex, or set enable_fused_normlization to False. I have installed it, but it is not working.
> 请使用 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: