RuntimeError: cuda runtime error (999) : unknown error at /opt/conda/conda-bld/pytorch_1595629403081/work/aten/src/THC/THCGeneral.cpp:47

[ptrblck]

You could try to reload the the nvidia kernel module via this approach.