Setup pytorch and torchvision on the Jetson

Reference: https://forums.developer.nvidia.com/t/pytorch-for-jetson-nano-version-1-5-0-now-available/72048/1