从源码编译MMCV

1. 安装 cuda

1. 本地环境：
显卡 RTX 3070 , 系统 Ubuntu 20.04，驱动 nvidia-driver-520，pytorch torch1.12.x
2. 驱动下载及安装
前往英伟达官方下载驱动 https://developer.nvidia.com/cuda-toolkit-archive（我使用了11.3）。下载方式使用 runfile(local)，以便进行后续的选择安装（不安装对应驱动）

sudo sh cuda_11.4.0_470.42.01_linux.run

在弹出的安装界面中选“continue”，下一步输入accept，

cuda安装包是自带显卡驱动的，所以这一步按空格去掉安装显卡驱动的选项，然后选择install。

成功后需要配置一下环境变量，打开.bashrc文件，配置环境

export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}

export export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

2. 下载mmcv源码并编译

直接按照深度眸大佬的来

git clone -b 2.x [email protected]:open-mmlab/mmcv.git 
cd mmcv

# 安装 ninja 和 psutil 以加快编译速度
pip install -r requirements/optional.txt

# 仅仅编译，不安装
MMCV_WITH_OPS=1 python setup.py build_ext --inplace

# 建立软链接
cd ../mmyolo
ln -s ../mmcv/mmcv mmcv

直接参考了大佬的文章进行了编译，一次成功，需要本地安装cuda驱动。
[1]: https://zhuanlan.zhihu.com/p/580885852
[2]: https://blog.csdn.net/weixin_44857882/article/details/124108458