CUDA runtime error (59) : device-side assert triggered

前端 未结 4 1797
星月不相逢
星月不相逢 2020-12-15 17:36

I have access to Tesla K20c, I am running ResNet50 on CIFAR10 dataset... Then I get the error as:
THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_152458471046

4条回答
  •  隐瞒了意图╮
    2020-12-15 18:01

    I encountered this error when running BertModel.from_pretrained('bert-base-uncased'). I found the solution by moving to the CPU when the error message changed to 'IndexError: index out of range in self'. Which led me to this post. The solution was to truncate sentences to length 512.

提交回复
热议问题