Блог

Я хочу использовать AWS Sagemaker для обучения модели PyTorch. Мне интересно, как мне назначить ресурсы для задачи? если бы у меня был свой собственный компьютер, я бы использовал:

 device = torch.cuda("cuda0")  
  

То же самое для AWS Sagemaker

Да, код устройства такой же на вашем собственном компьютере. Однако лучше всего сделать это

 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
  

Если вы хотите более подробно ознакомиться с PyTorch в Sagemaker, существует официальная документация https://sagemaker.readthedocs.io/en/stable/frameworks/pytorch/using_pytorch.html