device = "cuda" if torch.cuda.is_available() else "cpu"
print(device)

# create data
weight = 0.6
bias = 0.4
start = 0
end = 1
ste[......]

继续阅读