FairMOT

FairMOT运行出错

问题： RuntimeError: cuda runtime error (999) : unknown error at /opt/conda/conda-bld/pytorch_1591914880026/work/aten/src/THC/THCGeneral.cpp:47

解决办法： 其实是DCNv2编译的问题。

./make.sh # build ##通过

python testcpu.py # run examples and gradient check on cpu ##通过

python testcuda.py # run examples and gradient check on gpu ##出错，报错和上面一样

conda create -n FairMOT python=3.7

conda activate FairMOT

conda install pytorch1.2.0 torchvision0.4.0 cudatoolkit=10.0

cd ${FAIRMOT_ROOT}

pip install -r requirements.txt

git clone https://github.com/CharlesShang/DCNv2

cd DCNv2

./make.sh# build

( python testcpu.py # run examples and gradient check on cpu

python testcuda.py # run examples and gradient check on gpu )

cuda 安装

和链接不一样的是，我安装的10+7.6

卸载：

To uninstall the CUDA Toolkit, run the uninstall script in /usr/local/cuda-10.0/bin

cd /usr/local/cuda/bin

sudo ./uninstall_cuda_10.0.pl

sudo rm -rf cuda-10.0