Когда я запускаю этот код в файле Python, он выдает ошибку ошибки сегментации:
import torch
import numpy as np
model = torch.nn.Linear(10,1000)
data = torch.Tensor(np.zeros((50,10)))
model(data)
#Segmentation fault (core dumped)
Когда я запускаю тот же код в блокноте Jupyter, он работает нормально. Я даже могу увеличить размер данных до (1000000, 10), и он все еще работает в ноутбуке Jupyter, так что, похоже, у меня достаточно памяти. Когда я уменьшаю размер данных до (10, 10), код запускается в файле python.
У меня есть Python 3.6, torch = 1.3.0, операционная система Ubuntu 18.04
Я попытался использовать отладчик pdb и обнаружил, что ошибка сегментации происходит в строке
ret = torch.addmm(bias, input, weight.t())
в модуле torch.nn.functional.linear.
Я также попытался перезагрузить сервер, и это неничем не помог.
Как мне продолжить отладку этого?