RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)`
Robertatransformers에서 지원하는 Roberta를 기반으로 Korquad 데이터를 학습 중 입니다. 한국어를 학습하기 위해서 Multilingual를 지원하는 XLM-RoBERTa를 사용하도록 소스를 수정했습니다. 소스를 수정하고 run_squad.py를 수행하니 다음과 같은 에러가 발생했습니다. /pytorch/aten/src/THC/THCTensorIndex.cu:361: void indexSelectLargeIndex(TensorInfo, TensorInfo, TensorInfo, int, int, IndexType, IndexType, long) [with T = float, IndexType = unsigned int, DstDim = 2, SrcDim = 2, IdxDim = -..
Error
2020. 2. 19. 21:13