系统环境
Ubuntu 18.04 + gcc 7.5 + cuda 10.1 + cudnn
1. 查看cuda版本 cat /usr/local/cuda/version.txt
需要提前安装的包
1. Anaconda python=3.7
2. pip install pycocotools
3. pip install cpython
4. conda install pytorch=1.4 cudatoolkit=10.1 torchvision -c pytorch
不要安装 pytorch=1.5, 安装 1.4 的版本
否则会报错: site_packages/torch文件夹下面的 cpp_extension.py文件出错, Error compiled exit with status
最后
pip install -v -e .
不要落下 那一个点