PyTorch can't find CUDA

Question

I installed PyTorch 1.7.1 and CUDA v11.2 on my PC which has a GeForce GTX 1070. When I run the import torch command it works find but when I run the torch