pytorch自定义CUDA编译卡住

问题

torch.utils.cpp_extension.load()长时间卡住

原因分析

上一次编译异常中断，并且.lock文件没有清楚，导致本次编译一直等待上传编译结束（只有.lock文件清除后才能进行新的编译）。

解决方法

清除一下之前编译的内容

rm -rf ~/.cache/torch_extensions/py39_cu116/your_project

注意your_project要改成你的编译项目名称，py39_cu116要按照你的环境。