Hi,
You might need to specify the CUDA version explicitly along with PyTorch version while installing via Pip I guess. For example.
pip install torch==1.8.1+cu102 

You can find the differences on conda and pip installation on official PyTorch page 1.