Pytorch 编译cpp、cuda扩展时卡在import

具体表现就是import的时候不动了:
Pytorch 编译cpp、cuda扩展时卡在import_第1张图片
其中的Encoder就是需要cpp、cuda编译扩展的部分

此时如果停止运行,可以看到如下提示(jupyter notebook能看到,pycharm好像看不到):
Pytorch 编译cpp、cuda扩展时卡在import_第2张图片
可以看到是一直卡在了time.sleep这里,怀疑是死锁了。但是由于已经重启过,不太可能是有别的进程在占用。看触发的while语句,上锁是通过一个lock_file文件实现的。打印出该路径找到文件(其实就是放在缓存目录下):
Pytorch 编译cpp、cuda扩展时卡在import_第3张图片
有个名叫lock的文件。怀疑是之前编译意外中断导致lock文件还留在这里导致一直上锁。删除该编译缓存文件夹重新编译后,问题解决。

你可能感兴趣的:(debug)