set up error

Congratulations on your great work!

Could you please share the details of your "environment.yaml", including the versions of CUDA, PyTorch, and Python you're using? 

I'm running into some environment-related issues while trying to configure the project according to the README instructions.