search

open-mmlab / mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.
https://mmsegmentation.readthedocs.io/en/main/
Apache License 2.0
7.68k stars 2.53k forks source link

RuntimeError: sigmoid_focal_loss_forward_impl: implementation for device cuda:0 not found. #3683

Closed miaomaomiaomaoda closed 1 month ago

miaomaomiaomaoda commented 1 month ago

I didn't have a problem with cpu training, but I ran into this problem with GPU training using the following commands python ../tools/train.py ./checkpoints/solov2_test.py

Here is my output：

info.txt

I tried #2250 , setting the FORCE_CUDA and MMCV_WITH_OPS before installing mmcv but it didn't work.

How can I solve this problem, thank you very much.