make test failed with ImportError: libtorch_cuda_cu.so: cannot open shared object file: No such file or directory

[OyamingO]

Environment：
pytorch: 1.8.0
cudatoolkit: 10.2.89
gcc: 10.1

[heikeyuhuajia]

Environment： pytorch: 1.8.0 cudatoolkit: 10.2.89 gcc: 10.1

I completed the installation and testing by upgrading pytorch.