MMCV源码编译

MMCV源码编译

下载源码

git clone --recursive --branch v1.4.1 https://github.com/open-mmlab/mmcv.git

设置环境变量

export CUDA_HOME
export CUDA_PATH=$CUDA_HOME
export MMCV_WITH_OPS=1
export TORCH_CUDA_ARCH_LIST="7.5"

编译安装

cd mmcv
python setup.py build_ext
python setup.py develop