问题

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

原因分析

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

解决方法

清除一下之前编译的内容

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

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

更多推荐