Torchvision install download. docker pull rocm/pytorch:latest .