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

最近研究语义分割,阅读了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’](改成双短线&#x

  • 12
    点赞
  • 53
    收藏
    觉得还不错? 一键收藏
  • 34
    评论
Win10是Windows操作系统的最新版本,它具有更好的性能和用户体验。Anaconda是一个流行的Python发行版,它内置了许多科学计算和数据分析的库。PyTorch是一个深度学习框架,可以进行神经网络的设计和训练。 在Windows 10上安装Anaconda和PyTorch是非常简单的。首先,我们可以从Anaconda官网下载适用于Windows 10的安装文件。下载完成后,击运行安装程序,并按照提示进行安装。安装过程中,我们可以选择安装Anaconda所需的库和工具。 安装完成后,我们可以打开Anaconda Navigator,这是一个可视化的管理界面,可以帮助我们轻松管理和使用各种Python包和环境。在Navigator中,我们可以创建新的Python环境,选择安装所需的包。 接下来,我们需要安装PyTorch。在Anaconda Navigator中,我们可以通过选择我们创建的Python环境,然后点击“Channels”选项卡,选择“conda-forge”通道。然后,在搜索框中输入“pytorch”,选择合适的版本进行安装。 等待安装完成后,我们就可以在Python环境中使用PyTorch进行深度学习的开发了。我们可以使用PyTorch构建神经网络模型,进行训练和预测。此外,Anaconda还提供了许多其他有用的库,如NumPy、Pandas和Matplotlib,这些库可以帮助我们进行数据处理和可视化。 总结来说,使用Win10操作系统安装并使用Anaconda和PyTorch是非常方便和简单的。通过Anaconda,我们可以快速配置Python环境,并方便地安装和管理所需的库和工具。而PyTorch则为我们提供了一个强大的深度学习框架,可以支持各种神经网络的设计和训练

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 34
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值