minimind
minimind copied to clipboard
关于训练过程中CPU的占用问题?
我的配置: 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 命令看到效果如下:
看起来确定是在GPU上运行起来
但是我看CPU处理器占用全部打到了一个内核中,请问这种情况的具体原因可能是什么?
这里有可能把任务分发到不同的CPU内核中运行吗?