当我试图运行一个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。谢谢
目前没有回答
相关问题 更多 >
编程相关推荐