selamikarakas

While trying to install with Conda I have noticed that environment.yml automatically installs the cpu versions of Pytorch and Torchvision. I wonder which verison should I install for cuda enabled...