如果我不转移张量和模型使用（设备），GPU仍将用于培训吗？
我正在使用Google Colab，如果我不这样做，我需要知道它是否使用任何GPU进行培训。
I am using google colab, and I need to know if it uses any GPU for training if I don't do model.to('cuda') and data.to('cuda')?
如果你对这篇内容有疑问，欢迎到本站社区发帖提问 参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。
绑定邮箱获取回复消息
由于您还没有绑定你的真实邮箱，如果其他用户或者作者回复了您的评论，将不能在第一时间通知您！
发布评论
评论（1）
如果您不使用
model.to（torch.device（'cuda''）））和dato.to（torch.device（'cuda''））您的模型和所有模型您的张量将留在默认设备上，该设备为 cpu ，因此他们不了解 gpu 的存在。 Pytorch使用 cpu 进行工作。您可以看到此 。
If you do not use
model.to(torch.device('cuda'))anddata.to(torch.device('cuda'))your model and all of your tensors will be remaining on default device which is CPU, so they do not understand the existence of GPU. PyTorch uses CPU for its work.You can see this Link for more information about torch.device.