pycharm tensor 从cpu转至gpu 启用梯度

import torch

# 检查CUDA是否可用
if torch.cuda.is_available():
    print("CUDA is available. GPU is", "ON")
    print("Number of GPUs available:", torch.cuda.device_count())
    print("Names of available GPUs:", [torch.cuda.get_device_name(i) for i in range(torch.cuda.device_count())])
else:
    print("CUDA is not available. GPU is", "OFF")


tensor = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32)
tensor.to('cuda')
# x = torch.randn(5, 3).to('cuda')  # 或者使用x = torch.randn(5, 3).cuda()
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
x = torch.tensor([1.0, 2.0, 3.0], device=device)
print("Device:", x.device)  # 设备
  
print("Requires Grad:", x.requires_grad)  # 是否启用梯度
x= torch.tensor(2.0, requires_grad=True)
print("Requires Grad:", x.requires_grad)  # 是否启用梯度