{"library":"torch","install":[{"cmd":"pip install torch","imports":["# For trusted checkpoints (your own models):\nmodel.load_state_dict(torch.load('model.pt', weights_only=True))\n\n# For checkpoints with non-tensor objects (optimizer states, custom classes):\ncheckpoint = torch.load('checkpoint.pt', weights_only=False)  # only for trusted files","device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')\nmodel = MyModel().to(device)\ntensor = tensor.to(device)"]},{"cmd":"pip install torch --index-url https://download.pytorch.org/whl/cu128","imports":[]},{"cmd":"pip install torch --index-url https://download.pytorch.org/whl/cu124","imports":[]},{"cmd":"pip install torch --index-url https://download.pytorch.org/whl/rocm6.2","imports":[]}]}