Closed
Description
Add Link
Multiple links - see below.
Describe the bug
For example: You should use torch.rand(device='cuda')
instead of torch.rand(device=torch.device('cuda'))
Instances of this issue:
- amp_recipe.py
- polynomial_autograd.py change to
device = "cuda" if torch.cuda.is_available() else "cpu"
on L21 - dcgan_faces_tutorial.py
- tuning_guide.py
- nestedtensor.py
- cpp_extension.py
- polynomial_tensor.py
- neural_style_tutorial.py
Describe your environment
- PyTorch 2.0