RuntimeError: NCCL error in: /opt/conda/conda-bld/pytorch_1603729138878/work/torch/lib/c10d/ProcessGroupNCCL.cpp:31, unhandled cuda error, NCCL version 2.7.8

[liutianyi00]

请问如何解决呢，我的cuda是12.4，其他包的版本都是按照您github上的版本安装的

[yuhongtian17]

12.4的nvcc cuda对mmrotate-0.3.3/0.3.4来说太高了。请先降低cuda版本，把rotated faster rcnn r50跑通。我们STD这个仓库代码README所述的环境配置是在2080Ti上进行的，可能与更先进的GPU存在兼容性问题。这是一个在3090/4090/A100/H100 GPU服务器上均已验证可行的环境配置中文说明，供您参考。