"ti dico, amico, dovresti semplicemente abbandonare pytorch. fallo tutto in cuda che hai scritto tu stesso"