CUDA error

[liyajun299576]

I am running your code, but there is a bug that I can't solve, I hope I can get your help, thank you.

RuntimeError: cublas runtime error : resource allocation failed at /opt/conda/conda-bld/pytorch_1556653099582/work/aten/src/THC/THCGeneral.cpp:228
terminate called after throwing an instance of 'c10::Error'
what(): CUDA error: an illegal memory access was encountered (insert_events at /opt/conda/conda-bld/pytorch_1556653099582/work/c10/cuda/CUDACachingAllocator.cpp:564)
frame #0: c10::Error::Error(c10::SourceLocation, std::string const&) + 0x45 (0x7ff0be23ddc5 in /home/vislab/anaconda3/lib/python3.6/site-packages/torch/lib/libc10.so)
frame #1: + 0x14792 (0x7ff0b8da5792 in /home/vislab/anaconda3/lib/python3.6/site-packages/torch/lib/libc10_cuda.so)
frame #2: c10::TensorImpl::release_resources() + 0x50 (0x7ff0be22d640 in /home/vislab/anaconda3/lib/python3.6/site-packages/torch/lib/libc10.so)
frame #3: + 0x3067fb (0x7ff0b94c47fb in /home/vislab/anaconda3/lib/python3.6/site-packages/torch/lib/libtorch.so.1)
frame #4: + 0x13ff1b (0x7ff0ed259f1b in /home/vislab/anaconda3/lib/python3.6/site-packages/torch/lib/libtorch_python.so)
frame #5: + 0x3bf384 (0x7ff0ed4d9384 in /home/vislab/anaconda3/lib/python3.6/site-packages/torch/lib/libtorch_python.so)
frame #6: + 0x3bf3d1 (0x7ff0ed4d93d1 in /home/vislab/anaconda3/lib/python3.6/site-packages/torch/lib/libtorch_python.so)

frame #21: __libc_start_main + 0xf0 (0x7ff0f4409830 in /lib/x86_64-linux-gnu/libc.so.6)