CasA icon indicating copy to clipboard operation
CasA copied to clipboard

关于多卡训练!!!

Open Karkers opened this issue 9 months ago • 2 comments

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

Karkers avatar May 04 '24 07:05 Karkers

/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 )

yu-tnt avatar May 09 '24 03:05 yu-tnt

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

Karkers avatar May 11 '24 02:05 Karkers