Hi,
Please follow the below instructions to install PyTorch and TorchVision with CUDA support:
Thanks.