我如何使用自己的 GPU 进行训练

我不确定默认情况下 u2net_train.py 是否使用我的 GPU,我如何确定?如果训练不使用 GPU,我是否可以更改?谢谢。