PyTorchはcpuを使ってモデルの演算方式をロードします。

339 ワード

PyTorch cpu モデルをロード

gpuがcudaサポートしていない時はモデルをロードしてcpuに計算します。
将


model = torch.load(path, map_location=lambda storage, loc: storage.cuda(device))

に変更する


model = torch.load(path, map_location='cpu')

その後、すべての変数の後ろにある.cuda()メソッドを削除します。
以上のPyTorchはcpuを使って模型の演算方式をロードします。つまり、小编を使って皆さんの全部の内容を共有します。