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

2024-09-21 05:47:50 发布

您现在位置:Python中文网/ 问答频道 /正文

当我试图运行一个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是一个列表,s是一个int数,keep\u im是一个火炬长传感器在

奇怪的是,当我把代码修改为

^{pr2}$

它在try中返回错误,然后在except中再次操作,并且成功。。。。在

有人知道这里发生了什么吗? 我在Ubuntu16中使用python2.7+pytorch 0.3+cuda8+cudnn7.1,Titan XP。谢谢


Tags: 代码httpsgithubcompytorchcondafilefail

热门问题