2021-03-20

最新推荐文章于 2026-04-02 09:20:00 发布

原创最新推荐文章于 2026-04-02 09:20:00 发布 · 532 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

深度学习

本文介绍如何在Windows 10系统上安装配置DCNv2，包括下载PyTorch 1.7源码、修改cpp_extension.py文件以适配编译环境，并通过python setup.py build develop完成安装过程。

win10+CUDA11.1 + python3.7 + pytorch1.7 安装DCNv2

1, 下载 PyTorch1.7的源码： https://github.com/lbin/DCNv2/tree/pytorch_1.7

2，将源码“Pycharm\Anaconda\Lib\site-packages\torch\utils\cpp_extension.py”中compiler_info.decode().strip()修改成compiler_info.decode(‘gbk’).strip() (不修改也能编译通过，都会提示“Pycharm\Anaconda\lib\site-packages\torch\utils\cpp_extension.py:287: UserWarning: Error checking compiler version for cl: [WinError 2] 系统找不到指定的文件。warnings.warn('Error checking compiler version for {}: {}'.format(compiler, error)))”，

源码command = [‘ninja’, ‘-v’]保持不变（此处注意要保持不变，改成‘--v’反而编译不通过）

3，运行python setup.py build develop，显示如下信息则编译成功：

copying build\lib.win-amd64-3.8\_ext.cp38-win_amd64.pyd -> Creating d:\pycharm\anaconda\lib\site-packages\DCNv2.egg-link (link to .) DCNv2 0.1 is already the active version in easy-install.pth

Installed d:\test\downprj\sr\dcnv2-pytorch_1.7\dcnv2-pytorch_1.7 Processing dependencies for DCNv2==0.1 Finished processing dependencies for DCNv2==0.1
参考：