nanodet+win10+cpu测试，GPU训练

CPU可以测试，但是不可以训练，建议一开始就直接下cuda_gpu版本的 。。。
https://github.com/RangiLyu/nanodet

**

一、测试CPU

**
如果要测试CPU版本，可以去pytorch官网下载对应版本的torch,我pip的是这个版本的。建议先安torch，再按照requirements.txt下载。

pip install torch==1.8.0+cpu torchvision==0.9.0+cpu torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

然后将demo.py中的device：“cuda:0”均改为“cpu",注释掉报错的cuda检测时间的语句（one_stage_detector.py）

# torch.cuda.synchronize()

就可以测试运行了

python demo/demo.py image --config config/nanodet-m.yml --model nanodet_m.pth --path demo/1.jpg

如果报错：ModuleNotFoundError: No module named ‘nanodet’
说明忘记了 python setup.py develop
在代码运行前develop一下就好。

如果报错：FileNotFoundError: [Errno 2] No such file or directory: ‘nanodet_m.pth’
下载对应模型放到相应位置就好

（图片没调整大小）

二、GPU训练

卸载CUDA和pytorch、tensorflow

先贴一张对应的版本图，省的安装完毕还得卸载重装

1、参考网站安装对应版本CUDA。
我安装的是11.0.2。安装好后测试：nvcc -V
（如果不成功可能是需要配置环境变量，参考博客）

2、安装cuDNN
去官网下载配套的cuDNN，具体过程参考博客。

3、安装torch（注意要和cuda版本一致）。我从网站下载的是
cu110/torch-1.7.0%2Bcu110-cp36-cp36m-win_amd64.whl
cu110/torchvision-0.8.1%2Bcu110-cp36-cp36m-win_amd64.whl
然后再安装

pip install torch-1.7.0+cu110-cp36-cp36m-win_amd64.whl
pip install torchvision-0.8.1+cu110-cp36-cp36m-win_amd64.whl

上图参考链接：https://blog.csdn.net/qq_18483627/article/details/105885483

4、再安装tensorflow-gpu。先查看匹配版本，再下载对应版本，并pip。
下载链接：（根据自己下载的版本调整网址）
https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.4.0-cp36-cp36m-win_amd64.whl