"Tô te falando, cara, você devia largar o Pytorch. Faça tudo em CUDA que você mesmo escreveu"