Recommended configuration: Python 3.7,PyTorch 1.4, CUDA 10.1 and mmcv 0.6.2.
conda create -n aLRPLoss python=3.7 -y
conda activate aLRPLoss
conda install pytorch=1.4 cudatoolkit=10.1 torchvision -c pytorch
pip install mmcv==0.6.2
git clone https://github.com/kemaloksuz/aLRPLoss.git
cd aLRPLoss
pip install -r requirements/build.txt
pip install "git+https://github.com/open-mmlab/cocoapi.git#subdirectory=pycocotools"
pip install -v -e .
如果没有一次成功可能会出现这个问题:
libtorch_cpu.so: cannot open shared object file: No such file or directory
解决办法:
rm -r build
python setup.py develop