RuntimeError: Ошибка CUDA: изображение ядра не доступно для выполнения на устройстве после модели.cuda () ⇐ Python

[Anonymous]

Я работаю над этой моделью: < /p>
class Model(torch.nn.Module):
def __init__(self, sizes, config):
super(Model, self).__init__()

self.lstm = []
for i in range(len(sizes) - 2):
self.lstm.append(LSTM(sizes, sizes[i+1], num_layers=8))
self.lstm.append(torch.nn.Linear(sizes[-2], sizes[-1]).cuda())
self.lstm = torch.nn.ModuleList(self.lstm)

self.config_mel = config.mel_features

def forward(self, x):
# convert to log-domain
x = x.clip(min=1e-6).log10()

for layer in self.lstm[:-1]:
x, _ = layer(x)
x = torch.relu(x)

#x = torch_unpack_seq(x)[0]

x = self.lstm[-1](x)
mask = torch.sigmoid(x)

return mask
< /code>
, а затем: < /p>
model = Model(model_width, config)
model.cuda()
< /code>
Но я получаю эту ошибку: < /p>
File "main.py", line 29, in
Model.train(args)
File ".../src/model.py", line 57, in train
model.cuda()
File ".../.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 637, in cuda
return self._apply(lambda t: t.cuda(device))
File ".../.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 530, in _apply
module._apply(fn)
File "/.../.local/lib/python3.8/site-packages/torch/nn/modules/module.py", line 530, in _apply
module._apply(fn)
File ".../.local/lib/python3.8/site-packages/torch/nn/modules/rnn.py", line 189, in _apply
self.flatten_parameters()
File ".../.local/lib/python3.8/site-packages/torch/nn/modules/rnn.py", line 175, in flatten_parameters
torch._cudnn_rnn_flatten_weight(
RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
< /code>
Я понятия не имею, почему это происходит. Я пытаюсь продвинуть модель и входы в CUDA, и я понимаю, была ли ошибка из -за некоторых моделей в ЦП и некоторыми в графическом процессоре. Но это не так. Я обнаружил некоторое решение для установки PIP здесь: Pytorch Cuda Ошибка: изображение ядра не доступно для выполнения на устройстве на RTX 3090 с CUDA 11.1
Но я не могу использовать его, так как я пытаюсь выполнять работу в удаленном репозиции, где у меня нет доступа к установке Pip.>

Подробнее здесь: https://stackoverflow.com/questions/699 ... n-the-devi