RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [256]] is at version 3; expected version 2 instead

[parisafarboud]

I got Runtime error in backward while running the code with any versions of python and pytorch, how should I solve that, and which versions satisfy?

[tgyhhh]

Hello, have you solved this problem?

[parisafarboud]

Hello, have you solved this problem?

Hi, yes I've solved the problem. The problem was because I was training on single GPU and I commented the line below:
wetr = DistributedDataParallel(wetr, device_ids=[args.local_rank], find_unused_parameters=True)

[WQH-52]

Hello, how can I solve this problem? I am also training on a single GPU. @parisafarboud