RuntimeError: cuda runtime error (78) : a PTX JIT compilation failed at

[Tudouu]

Wonderful code!
Howerever,I encountered the following error.Did I miss something that caused this error?

mode : predcls
save_path : /media/wow/disk2/AG/save
model_path : /media/wow/disk2/AG/predcls.tar
data_path : /media/wow/disk2/AG/dataset
datasize : large
ckpt : None
optimizer : adamw
lr : 1e-05
nepoch : 10
enc_layer : 1
dec_layer : 3
bce_loss : False
-------loading annotations---------slowly-----------
--------------------finish!-------------------------
xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
There are 1750 videos and 56923 valid frames
41 videos are invalid (no person), remove them
19 videos are invalid (only one frame), remove them
8636 frames have no human bbox in GT, remove them!
xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
loading word vectors from data/glove.6B.200d.pt
loading word vectors from /media/wow/disk2/AG/glove.6B.200d.pt
background -> background
fail on background

---

CKPT /media/wow/disk2/AG/predcls.tar is loaded
THCudaCheck FAIL file=/home/cong/Dokumente/faster-rcnn.pytorch/lib/model/csrc/cuda/ROIAlign_cuda.cu line=297 error=78 : a PTX JIT compilation failed
Traceback (most recent call last):
File "test.py", line 80, in
entry = object_detector(im_data, im_info, gt_boxes, num_boxes, gt_annotation, im_all=None)
File "/home/wow/anaconda2/envs/STTran/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, **kwargs)
File "/media/wow/disk2/STT/STTran-main/lib/object_detector.py", line 306, in forward
FINAL_FEATURES = self.fasterRCNN.RCNN_roi_align(FINAL_BASE_FEATURES, FINAL_BBOXES)
File "/home/wow/anaconda2/envs/STTran/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call
result = self.forward(*input, **kwargs)
File "/media/wow/disk2/STT/STTran-main/fasterRCNN/lib/model/roi_layers/roi_align.py", line 58, in forward
input, rois, self.output_size, self.spatial_scale, self.sampling_ratio
File "/media/wow/disk2/STT/STTran-main/fasterRCNN/lib/model/roi_layers/roi_align.py", line 20, in forward
output = _C.roi_align_forward(input, roi, spatial_scale, output_size[0], output_size[1], sampling_ratio)
RuntimeError: cuda runtime error (78) : a PTX JIT compilation failed at /home/cong/Dokumente/faster-rcnn.pytorch/lib/model/csrc/cuda/ROIAlign_cuda.cu:297

[Tudouu]

I complied by the three 'python setup.py build_ext --inplace' from two readme in the project.I met this error when I wanted to evaluate the STTran.

[yrcong]

I have directly uploaded the local code which is already compiled on my computer.. It seems the issue is caused by the compiled code. Could you download the roi align module from https://github.com/jwyang/faster-rcnn.pytorch/tree/pytorch-1.0 and update the roi align file. Then compile it.

[Tudouu]

I have directly uploaded the local code which is already compiled on my computer.. It seems the issue is caused by the compiled code. Could you download the roi align module from https://github.com/jwyang/faster-rcnn.pytorch/tree/pytorch-1.0 and update the roi align file. Then compile it.

Thanks for your reply.I deleted those files which were generated from the compiling before and recompiled.The problem was solved then,it worked.