error in ms_deformable_col2im_cuda: invalid device function 报错

一句话总结:

使用你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 自己安装就好了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值