Wouldnt this be fixed by CUDA_LAUNCH_BLOCKING=1? Or putting a bunch of torch.cud... | Hacker News

Hacker Newsnew | past | comments | ask | show | jobs | submit

atorodius on Dec 15, 2020 | parent | context | favorite | on: What I wish someone had told me about tensor compu...

Wouldnt this be fixed by CUDA_LAUNCH_BLOCKING=1? Or putting a bunch of torch.cuda.synchronizes in the suspected lines.

whimsicalism on Dec 15, 2020 [–]

lol whoops yeah that would definitely solve the problem. I wasn't familiar with `CUDA_LAUNCH_BLOCKING` but `torch.cuda.synchronizes` does work.

Guidelines | FAQ | Lists | API | Security | Legal | Apply to YC | Contact