windows下mmdetection2d/3d gpu 環(huán)境搭建
我的電腦配置是mx130,384個(gè)cuda,4GB顯存,跑mmdetection推理速度還行。

cuda 11.1環(huán)境搭建
查看支持cuda版本,向下兼容,我安裝的cuda11.1,而驅(qū)動(dòng)已支持到了12.0

下載cuda11.1默認(rèn)安裝就好
https://developer.nvidia.com/cuda-11.1.1-download-archive
驗(yàn)證安裝在命令行里輸入nvcc -V

cuddn下載鏈接在此
https://developer.nvidia.cn/zh-cn/cudnn
將解壓后的內(nèi)容拷貝到cuda 安裝路徑:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
至此cuda環(huán)境已搭建完畢。
conda 配置國內(nèi)源
github加速訪問

mmdetection2d

如果第五步?jīng)]有因?yàn)椤盡icrosoft Visual C++ 14.0 or greater is required“安裝失敗,那么大功告成??! 若失敗參考則配置C++環(huán)境后再重新執(zhí)行這一步,參考此鏈接解決https://zhuanlan.zhihu.com/p/471661231


mmdetection3d

若運(yùn)行時(shí)遇到numba錯(cuò),需重新安裝numpy,參考此鏈接https://blog.csdn.net/qq_28668779/article/details/129238360