layumi / person-reid-3d

TNNLS'22 :statue_of_liberty: Parameter-Efficient Person Re-identification in the 3D Space :statue_of_liberty:
https://arxiv.org/abs/2006.04569
MIT License
264 stars 46 forks source link

请问能使用pytorch1.8吗? #19

Open MarklearnML opened 1 year ago

MarklearnML commented 1 year ago

如题,因为我只有30系列的显卡,但30系列好像只支持cuda11.1以上的版本,然而cuda11.1只有在pytorch1.7.1以上版本才支持。看到主页写着pytorch=1.4,所以想问一下该代码是否也能用pytorch1.8执行,非常感谢

layumi commented 1 year ago

requirements 中的 这个 https://github.com/layumi/person-reid-3d/blob/master/requirements.txt#L6 我估计 1.8 可能报错。

cuda 11的话 ,唯一要改的是 pointnet2_ops_lib。

  1. 安装 pointnet2_ops_lib (cuda11版本的)
  2. 把我里面一些修改的函数贴过去。就行了。
layumi commented 1 year ago

安装 pointnet2_ops_lib (cuda11版本的) 可以参考 这个
https://github.com/erikwijmans/Pointnet2_PyTorch
但我不确定有没有兼容 cuda11。 可以先试试 。

或者你自己搜一下github,应该能找到 cuda11的pointnet2_ops_lib 。

layumi commented 1 year ago

我修改的函数 。你可以对比一下。主要是在这两个 py 里。 其他都没动。

image

MarklearnML commented 1 year ago

好的 回答的很快 非常感谢!

ppsychoo commented 1 year ago

好的 回答的很快 非常感谢!

老哥,你用pytorch1.8跑成功了吗?我这边用dgl的时候一直说1.8与dgl不匹配。 如果跑通了,可以请教你一下 cuda11的pointnet2_ops_lib的问题吗?

MarklearnML commented 1 year ago

好的 回答的很快 非常感谢!

老哥,你用pytorch1.8跑成功了吗?我这边用dgl的时候一直说1.8与dgl不匹配。 如果跑通了,可以请教你一下 cuda11的pointnet2_ops_lib的问题吗?

@ppsychoo 跑通了 不过我不是用pytorch1.8, 我用的具体版本是pytorch=1.12.0 cuda=11.3 dgl=1.0.1cuda113.这样应该不会跟dgl冲突 pointnet2_ops_lib的问题我是通过这个解决的https://github.com/facebookresearch/votenet/issues/108#issuecomment-783878066 你就按照链接的教程安装官方的pointnet2,然后将该项目的pointnet2_modules.py和pointnet2_utils.py里修改过的代码复制到之前安装的pointnet2对应的文件里应该就行了。

ppsychoo commented 1 year ago

好的,谢谢你!

Psycho @.***

 

------------------ 原始邮件 ------------------ 发件人: "layumi/person-reid-3d" @.>; 发送时间: 2023年5月14日(星期天) 下午5:33 @.>; @.**@.>; 主题: Re: [layumi/person-reid-3d] 请问能使用pytorch1.8吗? (Issue #19)

好的 回答的很快 非常感谢!

老哥,你用pytorch1.8跑成功了吗?我这边用dgl的时候一直说1.8与dgl不匹配。 如果跑通了,可以请教你一下 cuda11的pointnet2_ops_lib的问题吗?

@ppsychoo 跑通了 不过我不是用pytorch1.8, 我用的具体版本是pytorch=1.12.0 cuda=11.3 dgl=1.0.1cuda113.这样应该不会跟dgl冲突 pointnet2_ops_lib的问题我是通过这个解决的facebookresearch/votenet#108 (comment) 你就按照链接的教程安装官方的pointnet2,然后将该项目的pointnet2_modules.py和pointnet2_utils.py里修改过的代码复制到之前安装的pointnet2对应的文件里应该就行了。

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>

ppsychoo commented 1 year ago

大佬关于pointnet2_ops_lib我还有点问题,可能加个QQ请教交流一下。我的QQ是2337537323。

好的 回答的很快 非常感谢!