Old cuda driver with Pytorch 1.5

I just ran

pip install torch

EDIT -
The problem got fixed when I switched drivers from proprietary 435 to 440 open source ones