search

ModelTC / United-Perception

United Perception
Apache License 2.0
426 stars 65 forks source link

多卡训练错误 #63

Closed hxy0307 closed 1 year ago

hxy0307 commented 1 year ago

ng=1时可以正常训练，但是ng>1时就会出错，想问问原版本的python，pytorch以及cuda的版本，我的错误如下：

WeChatcedc134115b20190ba1c9d6cdd5d3337
yqyao commented 1 year ago

pytorch1.8.1 cuda9.0 or cuda11.2 is ok, python3.6 @hxy0307