Hi,
It should work. Please follow the below command to install PyTorch:
And the below command for TorchVision and TorchAudio:
Thanks.