Traceback (most recent call last):


 "./pytorch/train.py", line 15, in <module>

from second.core import box_np_ops


 "/home/yyg/project/PointPillars/second/core/box_np_ops.py", line 7, in <module>

from second.core.non_max_suppression.nms_gpu import rotate_iou_gpu_eval


 "/home/yyg/project/PointPillars/second/core/non_max_suppression/__init__.py", line 2, in <module>

from second.core.non_max_suppression.nms_gpu import (nms_gpu, rotate_iou_gpu,


 "/home/yyg/project/PointPillars/second/core/non_max_suppression/nms_gpu.py", line 36, in <module>

@cuda.jit('(int64, float32, float32[:, :], uint64[:])')


 "/home/yyg/.conda/envs/torch_new/lib/python3.6/site-packages/numba/cuda/decorators.py", line 96, in kernel_jit

kernel.bind()


 "/home/yyg/.conda/envs/torch_new/lib/python3.6/site-packages/numba/cuda/compiler.py", line 489, in bind

self._func.get()


 "/home/yyg/.conda/envs/torch_new/lib/python3.6/site-packages/numba/cuda/compiler.py", line 370, in get

ptx = self.ptx.get()


 "/home/yyg/.conda/envs/torch_new/lib/python3.6/site-packages/numba/cuda/compiler.py", line 342, in get

**self._extra_options)


 "/home/yyg/.conda/envs/torch_new/lib/python3.6/site-packages/numba/cuda/cudadrv/nvvm.py", line 464, in llvm_to_ptx

libdevice = LibDevice(arch=opts.get('arch', 'compute_20'))


 "/home/yyg/.conda/envs/torch_new/lib/python3.6/site-packages/numba/cuda/cudadrv/nvvm.py", line 340, in __init__

raise RuntimeError(MISSING_LIBDEVICE_FILE_MSG.format(arch=arch))
RuntimeError: Missing libdevice file for compute_50.

Traceback (most recent call last):


 "train.py", line 192, in <module>

trainer.train()


 "train.py", line 41, in train

self._run_epoch(epoch)


 "train.py", line 75, in _run_epoch

loss_content = self.criterionG(outputs, targets)


 "/home/a/DeblurGANv2/models/losses.py", line 63, in __call__

return self.get_loss(fakeIm, realIm)


 "/home/a/DeblurGANv2/models/losses.py", line 54, in get_loss

fakeIm[0, :, :, :] = self.transform(fakeIm[0, :, :, :])


 "/home/a/DeblurGANv2/venv/lib/python3.6/site-packages/torchvision/transforms/transforms.py", line 163, in __call__

return F.normalize(tensor, self.mean, self.std, self.inplace)


 "/home/a/DeblurGANv2/venv/lib/python3.6/site-packages/torchvision/transforms/functional.py", line 208, in normalize

tensor.sub_(mean[:, None, None]).div_(std[:, None, None])
RuntimeError: expected type torch.cuda.FloatTensor but got torch.FloatTensor