Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
MarklearnML opened this issue Feb 15, 2023 · 8 comments
Open

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

MarklearnML opened this issue Feb 15, 2023 · 8 comments

Comments

@MarklearnML
Copy link

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

@layumi
Copy link
Owner

layumi commented Feb 15, 2023

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
Copy link
Owner

layumi commented Feb 15, 2023

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

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

@layumi
Copy link
Owner

layumi commented Feb 15, 2023

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

image

@MarklearnML
Copy link
Author

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

@ppsychoo
Copy link

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

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

@MarklearnML
Copy link
Author

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

老哥,你用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对应的文件里应该就行了。

@ppsychoo
Copy link

ppsychoo commented May 15, 2023 via email

@ppsychoo
Copy link

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

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants