2018-10-01から1ヶ月間の記事一覧

WSLでCuda toolkitがインストールできない問題について

WSLでCuda toolkitを入れようとして,このページを見ながらやっていたのですが, qiita.com

PyTorchのDataParallelのモデルを保存する

PyTorchで複数GPUで学習させる場合, model = nn.DataParallel(model, device_ids=[0,1,2]) のようにDataParallelで保存しますが,このモデルを保存したい場合にcuda runtime error : out of memoryが出ることがあります. その場合は,下のようにDataParall…