반응형
RuntimeError: t == DeviceType::CUDAINTERNAL ASSERT FAILED at "/opt/conda/conda-bld/pytorch_1634272068694/work/c10/cuda/impl/CUDAGuardImpl.h":24, please report a bug to PyTorch.
에러가 발생했다.
Pytorch Lightning을 사용하다 발생한 에러인데, torchmetric에서 불러온 PSNR 함수를 .cuda()로 보내지 않아 발생한 오류였다.
Pytorch Lightning이 torchmetric에서 불러온 PSNR 함수를 자동으로 cuda로 전달하지 않나보다.
psnr = PSNR().cuda()
반응형
'Python > PyTorch 공부' 카테고리의 다른 글
| VOC class name (1) | 2022.02.09 |
|---|---|
| [Pytorch] RuntimeError: Found dtype Long but expected Float (0) | 2021.12.18 |
| [Pytorch] Load state_dict로 인한 out of memory (0) | 2021.12.14 |
| [PyTorch] model.state.dict() 파라미터 확인하기 (0) | 2021.12.06 |
| [PyTorch] RuntimeError("grad can be implicitly created only for scalar outputs") (0) | 2021.12.03 |