模型训练时不使用gpu

[zhang13359]

我在三台windows电脑上训练，显卡分别是4060，1070，2080ti，训练时gpu占用率始终为0，gpu内存倒是占满了。在租的gpu云上，用4090在linux系统跑gpu占用就可以满。三台windows电脑的pytorch均可以使用gpu，不知道是什么原因。
上面4台电脑的cuda和pytorch版本均相同。
请问是什么回事

[o7451]

2080ti要跑多久，3000epochs 100000iters

[wangxinchao-bit]

可能是原始的数据集分辨率太大，从而加载数据集占据大部分时间，模型计算占据的时间较少