why not use gpu - Githubissues

LibCity / Bigscity-LibCity

LibCity: An Open Library for Urban Spatial-temporal Data Mining

https://libcity.ai/

Apache License 2.0

871 stars 159 forks source link

Open nana1223 opened 7 months ago

nana1223 commented 7 months ago

您好，想请问下，我在跑需求预测任务用DMVSTNet模型跑TAXIBJ数据集，但是却没有用到gpu，一直在用cpu跑，查看了代码中的配置确实是在使用gpu的呀，想请问下这种情况怎么去修改它命中gpu呢？（已确认本机有gpu且pytorch也是下的gpu版本的）感谢回复！

Blore-lzn commented 7 months ago

先看下torch.cuda.is_available()的结果是True吗

nana1223 commented 7 months ago

先看下torch.cuda.is_available()的结果是True吗

是true的