no kernel image is available for execution on the device at

[MIYU8305]

I did
python3 src/demo.py ctdet --demo images --load_model models/ctdet_coco_dla_2x.pth and
python3 src/demo.py multi_pose --demo images --load_model models/multi_pose_dla_3x.pth
Using centernet docker image

I got error
THCudaCheck FAIL file=/pytorch/aten/src/THCUNN/generic/Threshold.cu line=34 error=48 : no kernel image is available for execution on the device
Traceback (most recent call last):
File "src/demo.py", line 57, in
demo(opt)
File "src/demo.py", line 50, in demo
ret = detector.run(image_name)
File "/home/user/workdir/src/lib/detectors/base_detector.py", line 118, in run
output, dets, forward_time = self.process(images, return_time=True)
File "/home/user/workdir/src/lib/detectors/ctdet.py", line 30, in process
output = self.model(images)[-1]
File "/root/.pyenv/versions/3.6.9/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/user/workdir/src/lib/models/networks/pose_dla_dcn.py", line 473, in forward
x = self.base(x)
File "/root/.pyenv/versions/3.6.9/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/user/workdir/src/lib/models/networks/pose_dla_dcn.py", line 288, in forward
x = self.base_layer(x)
File "/root/.pyenv/versions/3.6.9/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/root/.pyenv/versions/3.6.9/lib/python3.6/site-packages/torch/nn/modules/container.py", line 91, in forward
input = module(input)
File "/root/.pyenv/versions/3.6.9/lib/python3.6/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/root/.pyenv/versions/3.6.9/lib/python3.6/site-packages/torch/nn/modules/activation.py", line 46, in forward
return F.threshold(input, self.threshold, self.value, self.inplace)
File "/root/.pyenv/versions/3.6.9/lib/python3.6/site-packages/torch/nn/functional.py", line 624, in threshold
return torch._C.nn.threshold(input, threshold, value)
RuntimeError: cuda runtime error (48) : no kernel image is available for execution on the device at /pytorch/aten/src/THCUNN/generic/Threshold.cu:34

How can I solve this?

[helloyuning]

Hello, MIYU8305.
I think your pytorch version mismatch the cuda version, maybe you can check the version and try it again.
Hope it can help

[Cutewr]

Have you solved this problem? I also encountered this problem.