minimind icon indicating copy to clipboard operation
minimind copied to clipboard

关于训练过程中CPU的占用问题?

Open mailguest opened this issue 1 day ago • 0 comments

我的配置: CPU: Intel(R) Xeon(R) CPU E5-2696 v3 * 2 GPU: NVIDIA GeForce RTX 4060 Ti 16GB * 1 RAM: DDR3 128GB

sentence-transformers 2.3.1 torch 2.6.0+cu126 torchaudio 2.6.0+cu126 torchvision 0.21.0+cu126 transformers 4.48.0

我在跑SFT的时候,确定是在GPU上运行了,当时通过 nvidia-smi 命令看到效果如下: Image 看起来确定是在GPU上运行起来

但是我看CPU处理器占用全部打到了一个内核中,请问这种情况的具体原因可能是什么? Image

这里有可能把任务分发到不同的CPU内核中运行吗?

mailguest avatar Feb 26 '25 07:02 mailguest