一句话总结:
使用你github上代码的原环境,就会没问题,没什么原因,就是pytorch torchvision cudatoolkit 的版本以及你显卡版本不匹配。同时DCN对不同的pytorch版本支持不同,验证下来,1.4肯定会有问题。1.2肯定没问题。其他高版本,看了网友留言多多少少都有点问题。
conda安装cudatoolkits
conda installcudatookit=10.X
当然你肯定要先看下和你的NVIDIA版本匹配不,或者说支持不。
pytorch 安装
https://pytorch.org/ 从这里找,没有的话,看previous 版本,然后如果是内网,就点开安装语句后边的链接,自己下在wheel的安装包,pip install 自己安装就好了。