The code is tested with Pytorch==1.12.0 and cuda 11.3 on V100 servers. To setup the python environment, follow: ...