Updated code for cuda 11 and torch 1.10+

[callzhang]

The code has been updated for newer versions of torch, cuda and other dependencies.

[Yanghojun]

@callzhang

Hi callzhang!

Thanks for your commit.

I want to operate your train code but I have one problem.

In your correlation.py code

ctx.save_for_backward(self, input1, input2)

return error in my desktop because save_for_backward function can save only Tensor. (self is Correlation class)

Did u solve this problem or Didn't you have any problem before?

[LongruiDong]

@callzhang hi I checkout your pr,but run into some error:
my python=3.6 pip=21.3.1 cuda= 11.4

ERROR: Could not find a version that satisfies the requirement cupy-cuda11x (from versions: none)
ERROR: No matching distribution found for cupy-cuda11x

ERROR: Invalid requirement: --no-build-isolation -e ./models/correlation_package
pip: error: no such option: --no-build-isolation

have u met those issues?

thanks in advance~

[LongruiDong]

@callzhang hi I checkout your pr,but run into some error: my python=3.6 pip=21.3.1 cuda= 11.4

ERROR: Could not find a version that satisfies the requirement cupy-cuda11x (from versions: none)
ERROR: No matching distribution found for cupy-cuda11x

ERROR: Invalid requirement: --no-build-isolation -e ./models/correlation_package
pip: error: no such option: --no-build-isolation

have u met those issues?

thanks in advance~

when I change python from 3.6 to 3.8，issue solved