环境:
cuda版本:Cuda compilation tools, release 10.0, V10.0.130
python:3.6.15
torch:1.4.0
torchvision:0.2.1
ninja:1.10.2
conda create -n FCOS python=3.6
conda activate FCOS
conda install ipython
pip install ninja==1.10.2 yacs cython matplotlib tqdm
pip install torch==1.4.0 torchvision==0.2.1 -i https://pypi.douban.com/simple
下载相关文件:
把下载的文件放在自己想放的地方就行,如我放在downloads下:cd home/Downloads
cocoapi:
git clone https://github.com/cocodataset/cocoapi.git
FCOS:
git clone https://github.com/tianzhi0549/FCOS.git
下面进行一系列编译:
cd home/Downloads/cocoapi/PythonAPI
python setup.py build_ext install
cd home/Downloads/FCOS
python setup.py build develop --no-deps