2GBかい、pytorchはメモリ抑えてくれる印象だがコードの書き方にもよるかも