Questions about python setup.py develop

[eggcat4399]

Hello! Thank you for your amazing work! I'm a newbie trying to reproduce and I get an error when running python setup.py develop:
(unitraj) C:\Windows\System32\UniTraj>python setup.py develop
Traceback (most recent call last):
File "C:\Windows\System32\UniTraj\setup.py", line 29, in
make_cuda_ext(
File "C:\Windows\System32\UniTraj\setup.py", line 8, in make_cuda_ext
cuda_ext = CUDAExtension(
File "C:\Users\Levi.conda\envs\unitraj\lib\site-packages\torch\utils\cpp_extension.py", line 1130, in CUDAExtension
library_dirs += library_paths(device_type="cuda")
File "C:\Users\Levi.conda\envs\unitraj\lib\site-packages\torch\utils\cpp_extension.py", line 1271, in library_paths
paths.append(_join_cuda_home(lib_dir))
File "C:\Users\Levi.conda\envs\unitraj\lib\site-packages\torch\utils\cpp_extension.py", line 2525, in _join_cuda_home
raise OSError('CUDA_HOME environment variable is not set. '
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
But for CUDA_HOME have already set up the environment and made sure that there is no problem, how do I solve it? Thankyou!