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

[Ming Wang]

Environment：
pytorch: 1.8.0
cudatoolkit: 10.2.89
gcc: 10.1

[Sherlock Wang]

Environment： pytorch: 1.8.0 cudatoolkit: 10.2.89 gcc: 10.1

I completed the installation and testing by upgrading pytorch.