Creating environment from yml file creates cpu only installations of torch etc.

Hello! Wonderful package. I just followed the installation files but running sample.py was taking me forever. I saw no memory used up on my GPU, and saw this when I ran conda list. I think why cpu only packages are being installed has to do with conda - I am about to try by either installing the packages manually since they are so few or using pip.

pytorch                   2.5.1              py3.12_cpu_0    pytorch
pytorch-cuda              11.7                 h778d358_5    pytorch
pytorch-mutex             1.0                         cpu    pytorch
torchvision               0.20.1                py312_cpu    pytorch