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

关于多卡训练!!! #31

Open
Karkers opened this issue May 4, 2024 · 2 comments
Open

关于多卡训练!!! #31

Karkers opened this issue May 4, 2024 · 2 comments

Comments

@Karkers
Copy link

Karkers commented May 4, 2024

大佬你好,在我的设备上pcdet0.3版本似乎不能进行多卡训练,运行script文件后就没有响应了,但是最新版本的pcdet能够进行多卡训练,我的设备是8*3090!请问你遇到过这个问题吗!

@yu-tnt
Copy link

yu-tnt commented May 9, 2024

/pcdet/utils/common_utils.py中的init_dist_pytorch函数初始的过程注释掉一部分参数。如下:
dist.init_process_group(
backend=backend,
# init_method='tcp://127.0.0.1:%d' % tcp_port,
# rank=local_rank,
# world_size=num_gpus
)

@Karkers
Copy link
Author

Karkers commented May 11, 2024

非常感谢,成功运行了!!!

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

2 participants