Skip to content

Err in 'PyTorch: Tensors' running on GPU #233

Closed
@planewave

Description

@planewave

# dtype = torch.device("cuda:0") # Uncomment this to run on GPU
it is supposed to be
# device = torch.device("cuda:0") # Uncomment this to run on GPU
same issue in PyTorch: Tensors and autograd

So x = torch.randn(N, D_in, device=device, dtype=dtype) is the new recommended style to create tensors?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions