a RuntimeError when run demo.py

[limengfei3675]

Hello, thanks for your code.My torchvision version is 0.4.2 and cuda version is 10.1. When I run demo.py I get a error : RuntimeError: CUDA error: no kernel image is available for execution on the device (nms_cuda at /tmp/pip-req-build-9d9zypi6/torchvision/csrc/cuda/nms_cuda.cu:127) frame #0: c10::Error::Error(c10::SourceLocation, std::__cxx11::basic_string<char, std::char_traits, std::allocator > const&) + 0x6d (0x7fabd233fe7d in /home/qxmz/anaconda3/envs/tensorflow1/lib/python3.7/site-packages/torch/lib/libc10.so) . It happened when nms in the code: anchors_nms_idx = nms(transformed_anchors[0, :, :], scores[0, :, 0], iou_threshold = self.iou_threshold) Can you tell me how to solve it??? Forgive me for poor english. Thank you very much. In addition, there were no problems with training.