У меня есть следующие строки кода (часть кода большего размера, но здесь я получаю сообщение об ошибке):
from torch.distributions import MultivariateNormal
prior = MultivariateNormal(torch.zeros(data.size(1)).cuda(), torch.eye(data.size(1)).cuda())
, и я получаю эту ошибку (для второй строки):
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
Я запустил точно такой же код на другой машине, и там он работает нормально, поэтому я не уверен, что не так. Кто-нибудь может мне помочь? Спасибо!