最近想跑Swin-transformer,看了下官方代碼,要使用到mmcv-segmentation這個庫。
但是使用windows自動安裝碰到了許多問題,記錄下解決方案和步驟。
首先安裝好Pytorch和cuda這個不必說。安裝好之后安裝mmcv
官方給的安裝方式是
- pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.1/index.html 但是在Windows下會各種報錯。
於是嘗試手動安裝。
1、裝vs的cl編譯器
2、github下載mmcv的源碼
3、安裝requirement
4、配置gpu算力
5、運行
python setup.py build_ext
python setup.py develop
參考網站:
https://blog.csdn.net/weixin_41922853/article/details/118445620
https://blog.csdn.net/qq_43209693/article/details/121558341?spm=1001.2101.3001.6650.1&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1.pc_relevant_default&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7ECTRLIST%7ERate-1.pc_relevant_default&utm_relevant_index=2
https://zhuanlan.zhihu.com/p/308281195