作者您好,最近在复现您的工作,我照着您所列出的环境要求配置好了环境,cuda和torch也是匹配的,但是运行train_reg.py文件时,遇到以下报错,不知是什么原因,期待您能够解答:
Traceback (most recent call last):
File "E:\fdr\UMF-CMGR\Trainer\train_reg.py", line 164, in
main(args, visdom)
File "E:\fdr\UMF-CMGR\Trainer\train_reg.py", line 83, in main
train(training_data_loader, optimizer, net, criterion, epoch, elastic, affine)
File "E:\fdr\UMF-CMGR\Trainer\train_reg.py", line 105, in train
ir_affine, affine_disp = affine(ir)
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
result = self.forward(*input, *kwargs)
File "E:\fdr\UMF-CMGR\functions\affine_transform.py", line 26, in forward
warped, affine_param = self.trs(input) # [batch_size, 3, 3]
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
result = self.forward(input, **kwargs)
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\kornia\augmentation\base.py", line 244, in forward
output = self.apply_func(in_tensor, in_transform, self._params, return_transform)
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\kornia\augmentation\base.py", line 204, in apply_func
output = self.apply_transform(in_tensor, params, trans_matrix)
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\kornia\augmentation\augmentation.py", line 719, in apply_transform
padding_mode=self.padding_mode.name.lower(),
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\kornia\geometry\transform\imgwarp.py", line 170, in warp_affine
dst_norm_trans_src_norm: torch.Tensor = normalize_homography(M_3x3, (H, W), dsize)
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\kornia\geometry\transform\homography_warper.py", line 378, in normalize_homography
src_pix_trans_src_norm = _torch_inverse_cast(src_norm_trans_src_pix)
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\kornia\utils\helpers.py", line 50, in _torch_inverse_cast
return torch.inverse(input.to(dtype)).to(input.dtype)
RuntimeError: CUDA error: no kernel image is available for execution on the device
作者您好,最近在复现您的工作,我照着您所列出的环境要求配置好了环境,cuda和torch也是匹配的,但是运行train_reg.py文件时,遇到以下报错,不知是什么原因,期待您能够解答: Traceback (most recent call last): File "E:\fdr\UMF-CMGR\Trainer\train_reg.py", line 164, in
main(args, visdom)
File "E:\fdr\UMF-CMGR\Trainer\train_reg.py", line 83, in main
train(training_data_loader, optimizer, net, criterion, epoch, elastic, affine)
File "E:\fdr\UMF-CMGR\Trainer\train_reg.py", line 105, in train
ir_affine, affine_disp = affine(ir)
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
result = self.forward(*input, *kwargs)
File "E:\fdr\UMF-CMGR\functions\affine_transform.py", line 26, in forward
warped, affine_param = self.trs(input) # [batch_size, 3, 3]
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\torch\nn\modules\module.py", line 722, in _call_impl
result = self.forward(input, **kwargs)
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\kornia\augmentation\base.py", line 244, in forward
output = self.apply_func(in_tensor, in_transform, self._params, return_transform)
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\kornia\augmentation\base.py", line 204, in apply_func
output = self.apply_transform(in_tensor, params, trans_matrix)
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\kornia\augmentation\augmentation.py", line 719, in apply_transform
padding_mode=self.padding_mode.name.lower(),
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\kornia\geometry\transform\imgwarp.py", line 170, in warp_affine
dst_norm_trans_src_norm: torch.Tensor = normalize_homography(M_3x3, (H, W), dsize)
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\kornia\geometry\transform\homography_warper.py", line 378, in normalize_homography
src_pix_trans_src_norm = _torch_inverse_cast(src_norm_trans_src_pix)
File "E:\fdr\conda\envs\UMF-CMGR\lib\site-packages\kornia\utils\helpers.py", line 50, in _torch_inverse_cast
return torch.inverse(input.to(dtype)).to(input.dtype)
RuntimeError: CUDA error: no kernel image is available for execution on the device