きょうも亀だけど

機械学習やデータサイエンスの勉強記録など

2020-10-28から1日間の記事一覧

PyTorchでCUDAを使ってGPUで学習をさせたときに「Expected all tensors to be on the same device, but found at least two devices」が出たときの対処

GPUを積んだローカルPCでPyTorchの転移学習を試した。以下を参考にCUDAでGPUを使用して学習させた。 www.amazon.co.jp 結果、Optimizerの実行箇所でタイトルのエラーが発生した。 「tensor使うなら全部GPUでやらなきゃいけないのにCPUが混ざってるぞ」と言わ…