Hi,
Do you want to use TorchVision with GPU support? Or CPU version is enough?
For CUDA version, please find the below page for more info:
Thanks.