device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
      Python language logo
      your recipe card header background

      Check Device

      0

      0

      avatar
      lucycodes42
      Pytorch Public Recipes

      Library: pytorch

      Shortcut: pytorch.device

      0 Comments

        Add Comment

        Log in to add a comment

        Codiga - All rights reserved 2022.