how to use gpu to train the model?

WuJie1010 / Facial-Expression-Recognition.Pytorch

A CNN based pytorch implementation on facial expression recognition (FER2013 and CK+), achieving 73.112% (state-of-the-art) in FER2013 and 94.64% in CK+ dataset

MIT License

1.78k stars 545 forks source link

how to use gpu to train the model? #74

Closed zhangzj8 closed 4 years ago

zhangzj8 commented 4 years ago

When I train the model, i use nvidia-smi to check the state of gpu,there are no gpu used.

WuJie1010 commented 4 years ago

please check your nvidia environment