2023-03

画像超解像復元の機械学習

PyTorchでGPU CUDAでCPU Offloadingした。

機械学習フレームワークのPyTorchにて、NVIDIA GeForce RTX 3060 GPU 12GB VRAMとメインメモリー64GBで、約3.65億パラメーターの大きなモデルのCPU Offloadingでのトレーニングに成功したので、そのやり方を掲載いたします。
私のPC環境

NVIDIA GeForce RTX 3060 GPU搭載グラフィクス カードを導入し、電力制限を設定した。

NVIDIA GeForce RTX 3060 GPU搭載グラフィクス カードをUbuntu OSな静音自作PCに搭載し、電力制限を設定した記録です。