使用异步调用cuda（）会导致语法错误

Question

我想运行这个PyTorch代码：

for i, (input, target) in enumerate(train_loader):

    input = input.float().cuda(async=True)
    target = target.cuda(async=True)
    input_var = torch.autograd.Variable(input)
    target_var = torch.autograd.Variable(target)

    output = model(input_var)

但当我尝试时，我收到了一条错误消息：

input = input.float().cuda(async=True)
                               ^
SyntaxError: invalid syntax
Process finished with exit code 1

我做错什么了？我已经安装了cuda。