pytorch installation in env.yaml without gpu support

[OliiverHu]

Hi!

Thanks for sharing this fantastic project. I encountered this problem during env configuration. It seems that following the tutorial does have everything set up smoothly. However, during training, I noticed that the torch is actually installed with the cpu version? (.cuda.is_available() = False). I am not sure what might cause the problem. I tried to uninstall torch and then re-install with a explictly stated gpu version but failed since the dependencies are incompatible.

Still working on it and we appreciate any insights you could provide!

Thanks!