Conda install torchvision pytorch. I am using Ubuntu 18.