PyTorchでTensorとモデルのGPU / CPUの切り替え

Question

オートエンコーダを実装する際に以下のようなエラーが発生してしまいます、解決策がわかる方がいましたらご教授お願いいたします。

Tensor for 'out' is on CPU, Tensor for argument #1 'self' is on CPU, but expected them to be on GPU (while checking arguments for addmm)

Answer 1

このコードだけではわかりません。

Tensorの演算で使用しているtorch.Tensor型のオブジェクトがtorch.cuda.FloatTensorとかtorch.cuda.IntTensorのようなGPU tensorでなければならないときにtorch.FloatTensorとかtorch.IntTensorのようなCPU tensorを使っているようです。

コード全体を調べてtorch.Tensor型のオブジェクトの定義で間違っているところを修正する必要があります。

参考

公式ドキュメント TORCH.TENSOR