WIN10+cuda10.2+pytorch1.1+VS2015+conda3-5.3.1
參考文檔：
總：https://github.com/YDS-FOREVER/CenterNet-windows/blob/master/readme/INSTALL.md
https://blog.csdn.net/sinat_29634715/article/details/102609876
https://blog.csdn.net/yangdashi888/article/details/103178870
- 下載安裝cuda
cuda: https://developer.nvidia.com/cuda-downloads
安裝完之后使用nvcc --version測試
如果找不到nvcc.exe, 需要將路徑加入PATH. (默認是: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin)
-
安裝visualcppbuildtools_full.exe（內網65對ftp也有）
聯網在線安裝，使用VPN進行
-
conda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
建議參考https://mirror.tuna.tsinghua.edu.cn/help/anaconda/ 進行國內進行鏡像的配置. -
進入虛擬環境
conda create -n CenterNet python=3.6
conda activate CenterNet
conda install pytorch==1.1.0 torchvision cudatoolkit=10.0 -c pytorch
git clone https://github.com/xingyizhou/CenterNet.git
cd CenterNet
pip install -r requirements.txt
4. 在線部署cocoapi
pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
5. 編譯DCNv2
git clone https://github.com/CharlesShang/DCNv2
#DCNv2文件夾替換CenterNet\src\lib\models\networks\DCNv2文件夾
#修改 DCN文件2\src\cuda\dcn_v2_cuda.cu
#//extern THCState *state; #注釋
##THCState *state = at::globalContext().lazyInitCUDA(); // 添加該行
cd CenterNet\src\lib\models\networks\DCNv2
python setup.py build develop
#可能需要下載模型，沒有的話