I just ran
pip install torch
EDIT -
The problem got fixed when I switched drivers from proprietary 435 to 440 open source ones
I just ran
pip install torch
EDIT -
The problem got fixed when I switched drivers from proprietary 435 to 440 open source ones