Hello! I have these lines of code (part of a larger code, but this is where I am getting the error):
from torch.distributions import MultivariateNormal
prior = MultivariateNormal(torch.zeros(data.size(1)).cuda(), torch.eye(data.size(1)).cuda())
and I am getting this error (for the second line):
File "/path/to/lib/python3.6/site-packages/torch/distributions/multivariate_normal.py", line 149, in __i\
nit__
    self._unbroadcasted_scale_tril = torch.cholesky(covariance_matrix)
RuntimeError: CUDA error: no kernel image is available for execution on the device

I ran the exactly same code on a different machine, and it works fine there, so I am not sure what is wrong. Can someone help me? Thank you!