Pytorchがcuda使ってるからvramでしか動かないんだと思う
このあたりなんとかすれば普通のramでもいけるようになるんじゃないか