运行时错误：在PyTorch中发生CUDA运行时错误（8）

Question

当我尝试运行来自https://github.com/rowanz/neural-motifs的PyTorch Faster RCNN代码时，我遇到了以下问题。

THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1518238409320/work/torch/lib/THC/generic/THCTensorMathPairwise.cu 
line=21 error=8 : invalid device function

当操作keep时，它会出现以下情况：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)

在try中出现错误，然后在except中重新运行并成功......

有人知道这里发生了什么吗？ 我使用Python 2.7 + PyTorch 0.3 + CUDA 8 + cuDNN 7.1，在Ubuntu 16上使用Titan XP。谢谢。

- user2481799

---

无效设备函数错误通常意味着CUDA代码未编译为与您的GPU兼容。 - talonmies

Answer 1

您需要升级torch版本。

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

**

- Yashi Aggarwal