"Я тебе говорю, чувак, тебе просто стоит отказаться от pytorch. Сделай всё на cuda, который ты сам написал"