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
posted @ 2022-03-22 18:18  nanmi  阅读(294)  评论(0)  编辑  收藏  举报