运行Demo出错

[megvii-basedetection]

运行Demo出错

[YanChenkai]

请把-c path/to/yolox_s.pth.tar换成自己电脑权重文件的所在地...

[liuwei3612]

[liuwei3612]

请把-c path/to/yolox_s.pth.tar换成自己电脑权重文件的所在地...

我改成相应的路径后，还是报这样的错误

[YanChenkai]

@liuwei3612 请大佬来解决吧，剩下的我不懂

如果正常的话，在打印出summary之后
| INFO | __main__:229 - Model Summary:Params: 8.97M, Gflops: 11.33
下一步应该是load模型参数，终端输出应该是
| INFO | __main__:240 - loading checkpoint

所以可以调试一下，查看一下demo.py中第229行到第240行之间哪里有问题？

还有我是在NVIDIA的docker里运行的，没有问题，如果实在没法解决要不要试一试？

[liuwei3612]

@liuwei3612 请大佬来解决吧，剩下的我不懂

如果正常的话，下一步应该是load模型参数，终端输出应该是
| INFO | __main__:240 - loading checkpoint

所以可以调试一下，查看一下demo.py中第229行到第240行之间哪里有问题？

还有我是在NVIDIA的docker里运行的，没有问题，如果实在没法解决要不要试一试？

谢谢

[Joker316701882]

@liuwei3612 Hi, does your machine have GPU?

[liuwei3612]

@liuwei3612 Hi, does your machine have GPU?

有的，只有一个

[Joker316701882]

Hi @liuwei3612 and all, we have updated demo.py. Many bugs are fixed. Please pull our latest branch!!
Now you can use --device cpu/gpu when running demo.py to specify the device (default CPU).

[liuwei3612]

Hi @liuwei3612 and all, we have updated demo.py. Many bugs are fixed. Please pull our latest branch!!
Now you can use --device cpu/gpu when running demo.py to specify the device (default CPU).

谢谢你的回复，目前同步到最新baranch，默认的device是CPU，可以运行成功。但改成成GPU，就出错了，错误信息如下：

目前我这边CUDA Version 10.1.243
工程运行的pytorch是什么版本啊，需要匹配的CUDA的版本是多少？

[Joker316701882]

pytorch==1.9.0, but I think as long as >=1.7.0, it should be fine.
cuda==10.2