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

Question

当我尝试运行 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 是一个 int 数字，keep_im 是一个 torch.cuda.Longtensor

奇怪的是，当我将代码修改为

try:
    keep.append(keep_im + s)
except BaseException:
    keep.append(keep_im + s)

它在 try 中返回错误，然后在 except 中再次操作，并成功......

有人知道这里发生了什么吗？我在 Ubuntu 16 中使用 python 2.7 + pytorch 0.3 + Cuda 8 + cudnn7.1、Titan XP。谢谢

Answer 1

您必须升级火炬版本。**

对我来说，升级到 torch 1.11.0 版本有效。

**