This code has been tested on on the workstation with Intel Xeon Silver 4210R CPU and GPUs of NVIDIA GeForce RTX 3090 with a single 24G of video memory, Python 3.8, pytorch 1.11.
All experiments are conducted on the machine with two GTX 3090 GPU and Intel(R) Xeon(R) Silver 4210R CPU @ 2.40GHz, and the code is implemented in Python.