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