когда я пытаюсь запустить Pycrch более быстрый код rcnn (из https://github.com/rowanz/neural-motifs), я получаю сообщение об ошибке следующим образом
THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1518238409320/work/torch/lib/THC/generic/THCTensorMathPairwise.cu
line=21 error=8 : invalid device function
Возникает, когда работает keep.append (keep_im + s), где keepэто список, s это целое число, а keep_im это torch.cuda.Longtensor
Странно, что когда я изменяю код как
try:
keep.append(keep_im + s)
except BaseException:
keep.append(keep_im + s)
, он возвращает ошибку при попытке изатем снова работает в исключениях, и успехи ....
Кто-нибудь знает, что здесь происходит? Я использую python 2.7 + pytorch 0.3 + Cuda 8 + cudnn7.1, Titan XP в Ubuntu 16. Спасибо