Open gnniaaa opened 3 years ago
When I try to run train.py, I find it run with CPU instead of GPU, how can I use GPU?
When I try to run train.py, I find it run with CPU instead of GPU, how can I use GPU?