How did you install pytorch on that particular machine?
Can you try in the command line to create a Tensor, send it to the GPU and try to pass it to torch.cholesky 1 ? To make sure this is the problematic function.