当我尝试运行pytorch更快的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)
它在尝试中返回错误,然后在异常中再次操作,并成功。...
有人知道这里会发生什么吗? 我在Ubuntu 16中使用python 2.7 + pytorch 0.3 + Cuda 8 + cudnn7.1,Titan XP。谢谢