System requirement: PyTorch version >=2.4.1, python == 3.10 Tested on GPUs: A800, python version == 3.10, PyTorch version == 2.4.1, cuda version == 12.1 ./test_data ...