显卡:NVIDIA GeForce RTX 3070 Laptop GPU
conda create -n mmseg python=3.7
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
pip install torch1.9.0
pip install torchaudio0.9.0
pip install torchvision==0.10.0
在环境变量配置好的情况下
$env:TORCH_CUDA_ARCH_LIST=“8.6”
$env:MMCV_WITH_OPS = 1
$env:MAX_JOBS = 8
python setup.py build_ext
python setup.py develop
注:执行python setup.py develop可能出现权限错误,用管理员身份打开Anaconda powershell prompt,再回到mmcv中执行python setup.py 就可以了develop
官方配置连接:https://mmcv.readthedocs.io/en/latest/get_started/build.html