Skip to content

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

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

Description

@planewave
planewave
opened on May 2, 2018

# 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?

Activity

  1. jxcodetw commented on May 3, 2018

    @jxcodetw
    jxcodetw
    on May 3, 2018
    Contributor

    Where did you see this line of code in the tutorial?
    Can you post some links?

  2. planewave commented on May 3, 2018

    @planewave
    planewave
    on May 3, 2018
    Author

    they are here:

    tutorials/beginner_source/examples_tensor/two_layer_net_tensor.py

    Line 26 in e55b6e2

    # dtype = torch.device("cuda:0") # Uncomment this to run on GPU

    tutorials/beginner_source/examples_autograd/two_layer_net_autograd.py

    Line 21 in e55b6e2

    # dtype = torch.device("cuda:0") # Uncomment this to run on GPU

  3. gloriouskilka commented on May 10, 2018

    @gloriouskilka
    gloriouskilka
    on May 10, 2018

    @jxcodetw Noticed it here https://pytorch.org/tutorials/beginner/pytorch_with_examples.html

  4. chsasank
    closed this as completedin 93e6ad2on May 19, 2018
  5. rodrigo-techera
    added a commit that references this issue on Nov 29, 2021

    Fix pytorch#233

    2a53e26
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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