ImportError: cannot import name 'IS_MLU_AVAILABLE' from 'mmcv.utils' (unknown location)

OpenGVLab / InternImage

[CVPR 2023 Highlight] InternImage: Exploring Large-Scale Vision Foundation Models with Deformable Convolutions

https://arxiv.org/abs/2211.05778

MIT License

2.52k stars 234 forks source link

ImportError: cannot import name 'IS_MLU_AVAILABLE' from 'mmcv.utils' (unknown location) #210

Closed MasahiroOgawa closed 1 year ago

MasahiroOgawa commented 1 year ago

I got the above error when I ran

$ CUDA_VISIBLE_DEVICES=0 python image_demo.py data/mine/00001.jpg configs/ade20k/upernet_internimage_t_512_160k_ade20k.py    checkpoint_dir/seg/upernet_internimage_t_512_160k_ade20k.pth    --palette ade20k

I think this is related to the mmcv version... I installed mmcv-full==1.5.0 as stated in segmentation/README.md

MasahiroOgawa commented 1 year ago

When I removed conda environment as;

conda remove -n internimage --all

And reinstall everything from scratch, the error disappeared.