PyTorch FAQ

约 1 分钟阅读

标签：FAQ

Apple MPS

python

import torch
if torch.backends.mps.is_available():
    mps_device = torch.device("mps")
    x = torch.ones(1, device=mps_device)
    print (x)
else:
    print ("MPS device not found.")

text

tensor([1.], device='mps:0')

torch dynamo

自动计算图生成和优化

最近更新2024年6月24日commit a33aca9 编辑

Pytorch Cookbook

Hub