yolact
yolact copied to clipboard
CPU version of PyTorch on environment.yml
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 Pytorch and Torchvision?