PVNet的demo报错解决方案(win10)

PVNet的demo报错解决方案:CUDA error: initialization error when calling cusparseCreate(handle)

主文件:PVNet的demo(win10)

win10下运行PVNet的demo时报错:

RuntimeError: CUDA error: initialization error when calling `cusparseCreate(handle)

报错全文如下:

Traceback (most recent call last):
  File "D:/E20220227/B015_00_PV_Net/tools/demo_feng.py", line 191, in <module>
    demo()
  File "D:/E20220227/B015_00_PV_Net/tools/demo_feng.py", line 177, in demo
    corner_pred = eval_net(seg_pred, vertex_pred).cpu().detach().numpy()[0]
  File "C:\Users\26735\anaconda3\envs\python367\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl
    return forward_call(*input, **kwargs)
  File "C:\Users\26735\anaconda3\envs\python367\lib\site-packages\torch\nn\parallel\data_parallel.py", line 166, in forward
    return self.module(*inputs[0], **kwargs[0])
  File "C:\Users\26735\anaconda3\envs\python367\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl
    return forward_call(*input, **kwargs)
  File "D:/E20220227/B015_00_PV_Net/tools/demo_feng.py", line 57, in forward
    return ransac_voting_layer_v3(mask, vertex_pred, 512, inlier_thresh=0.99)
  File "D:\E20220227\B015_00_PV_Net\lib\ransac_voting_gpu_layer\ransac_voting_gpu.py", line 600, in ransac_voting_layer_v3
    all_win_pts=torch.matmul(b_inv(ATA),torch.unsqueeze(ATb,2)) # [vn,2,1]
  File "D:\E20220227\B015_00_PV_Net\lib\ransac_voting_gpu_layer\ransac_voting_gpu.py", line 516, in b_inv
    b_inv, _ = torch.linalg.solve(eye, b_mat)
RuntimeError: CUDA error: initialization error when calling `cusparseCreate(handle)`

定位到报错位置b_inv, _ = torch.linalg.solve(eye, b_mat)进行debug,发现eye.shape=torch.Size([9, 2, 2]), device=cuda:0,b_mat.shape=torch.Size([9, 2, 2]), device=cuda:0,而报错是由于CUDA,所以先把eye和 b_mat转成cpu,再计算结果,再把结果转换为cuda。代码修改后如下:

b_inv, _ = torch.linalg.solve(eye.cpu(), b_mat.cpu()).cuda()

运行继续报错:

ValueError: too many values to unpack (expected 2)

这是因为等号右侧结果只有1个,而等号左侧输出有2个,因此,删除不需要的输出"_",代码修改后如下:

b_inv = torch.linalg.solve(eye.cpu(), b_mat.cpu()).cuda()

再次运行,通过!!输出了和作者一样的demo结果,开心()!!
PVNet demo 运行结果

待填坑:一个没发现的问题

绿色框表示猫的边界框真值,蓝色框表示预测值,但是图里只有绿色框,没有蓝色框,待我查一查这是为什么。。。。

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
clean-pvnet是一个用于6DoF姿态估计的代码库,它是基于“PVNet: Pixel-wise Voting Network for 6DoF Pose Estimation”论文开发的。如果您想要下载clean-pvnet代码并配置环境,可以按照以下步骤进行操作: 1. 使用命令`git clone https://github.com/zju3dv/clean-pvnet.git`下载代码。 2. 配置环境,可以使用以下命令: ``` conda create -n pvnet python=3.7 conda activate pvnet conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 -c pytorch pip install Cython==0.28.2 sudo apt-get install libglfw3-dev libglfw3 pip install -r requirements.txt ``` 这些命令将创建一个名为`pvnet`的虚拟环境,并安装所需的依赖项。 3. 当您运行`run.py`时,输出日志将显示运行的进度和结果。根据提供的日志,运行看起来是成功的,并输出了一些信息。 4. 对于权重文件,根据提供的信息,训练好的权重文件可能存储在代码库的某个目录中。 请注意,这只是对clean-pvnet的简要介绍,如果您需要更多详细信息或有其他相关问题,请参考论文和代码库中的文档。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* *3* [【复现笔记】clean-pvnet复现](https://blog.csdn.net/weixin_54470372/article/details/128293400)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 100%"] [ .reference_list ]

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值