Accidentally removed init_cuda call.