YOLOV3 报错 RuntimeError: Expected all tensors to be on the same device

使用的是U版的yolov3程序,运行了一个Epoch 到 test.py就报错

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

找了好多方法都不管用啊，直到我看见这个链接

https://github.com/cwq159/PyTorch-Spiking-YOLOv3/issues/16

应该是torch版本的问题

解决办法，卸载高版本的torch 1.8.1，安装1.3.1。

代码：

$ pip3 install torch===1.3.1 torchvision===0.4.2 -f https://download.pytorch.org/whl/torch_stable.html

至此问题解决，加油啊！