Win10+Pytorch1.0编译并训练语义分割网络DANet(双注意力机制)

最新推荐文章于 2025-05-20 10:01:44 发布

原创

最新推荐文章于 2025-05-20 10:01:44 发布 · 1.2w 阅读

53 ·

CC 4.0 BY-SA版权

文章标签：

#Pytorch #Win10

本文详细记录了在Win10系统中，使用Pytorch1.0编译并训练DANet（双注意力机制）语义分割网络的过程，包括遇到的Subprocess.CalledProcessError、找不到编译器路径、模块导入错误等问题，以及相应的解决办法。解决方法涉及到修改cpp_extension.py文件、更新环境变量、修改encoding模块代码等。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近研究语义分割，阅读了dual attention的文章，于是想尝试跑跑源码。结果过程异常艰辛，辛辛苦苦花了好几天时间编译，踩过的坑不计其数，不过还好最后编译成功。这里记录下几个主要的坑，希望能够帮助相关研究的同志们。
先说一下自己的平台和配置：
Win10, CUDA10, Pytorch1.0, python:3.7, Anaconda3, VS2017

DANet地址：https://github.com/junfu1115/DANet
作者经过多次更新，明确下我下载调试的时间2019年5月28日，此时作者使用的encoding模块版本是torch_encoding-0.4.5-py3.7。需要注意的是，encoding模块自身已经更新到1.0版本以上，我同时下载了encoding模块https://github.com/zhanghang1989/PyTorch-Encoding的代码，以备参考。注意不能直接替换DANet中的encoding模块版本，因为DANet作者进行了修改。

我是在Git Bash中运行命令行，首先根据Readme指示，在代码文件夹路径下进行python setup.py install操作。

1.报错Subprocess.CalledProcessError:Command ‘[‘ninja’,’-v’]’ returned non-zero exit status 1.
参考：https://zhuanlan.zhihu.com/p/53418563
解决方法是：修改…Anaconda3/Lib/site-packages/torch/utils/cpp_extension.py文件
将[‘ninja’,’-v’]改成[‘ninja’,’- -v’]（改成双短线）

2.subprocess.CalledProcessError: Command ‘[‘where’, ‘cl’]’ returned non-zero exit status 1.
这个问题是因为电脑安装的VS2017，没有将编译器路径加入环境变量。解决方法参考链接