解决方法

我这边最终的解决方案就是 发现PyTorch 的CUDA版本，以及自己装的cuda 版本不一致

我pytorch 11.7改成 11.6就好了。

这里大家需要make sure

1. pytorch cuda
2. nvidia-smi 里面显示的版本
3. nvcc 显示的版本
4. 如果你用docker，docker的cuda版本

这些都需要一致。我里面当时有的11.7, 有的11.6 就出问题了。