欢迎光临散文网 会员登陆 & 注册

windows下mmdetection2d/3d gpu 环境搭建

2023-03-01 16:57 作者:钢镚乘六  | 我要投稿

我的电脑配置是mx130,384个cuda,4GB显存,跑mmdetection推理速度还行。

cuda 11.1环境搭建

查看支持cuda版本,向下兼容,我安装的cuda11.1,而驱动已支持到了12.0

查看本机的cuda版本

下载cuda11.1默认安装就好

https://developer.nvidia.com/cuda-11.1.1-download-archive

验证安装在命令行里输入nvcc -V

nvcc -v

cuddn下载链接在此

https://developer.nvidia.cn/zh-cn/cudnn

将解压后的内容拷贝到cuda 安装路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1

至此cuda环境已搭建完毕。

conda 配置国内源

github加速访问

mmdetection2d


如果第五步没有因为”Microsoft Visual C++ 14.0 or greater is required“安装失败,那么大功告成!! 若失败参考则配置C++环境后再重新执行这一步,参考此链接解决https://zhuanlan.zhihu.com/p/471661231

demo

mmdetection3d

result

若运行时遇到numba错,需重新安装numpy,参考此链接https://blog.csdn.net/qq_28668779/article/details/129238360

windows下mmdetection2d/3d gpu 环境搭建的评论 (共 条)

分享到微博请遵守国家法律