如何解决在AWS Sagemaker上使用PyTorch时如何分配修饰符?
我希望使用AWS Sagemaker训练PyTorch模型。我想知道如何为任务分配资源?如果我有自己的计算机,我会使用:
device = torch.cuda("cuda0")
AWS Sagemaker是否相同
解决方法
是的,设备代码与您自己的计算机相同。但是更好的事情是
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
如果您想在Sagemaker上查找有关PyTorch的更多详细信息,请参阅官方文档https://sagemaker.readthedocs.io/en/stable/frameworks/pytorch/using_pytorch.html
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。