RuntimeError:pytorch中的cuda运行时错误(8)

时间:2018-10-18 09:39:45

标签: python pytorch

当我尝试运行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。谢谢

0 个答案:

没有答案