RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM

596 ワード

問題の説明:
入力データをネットワークモデルに送ったときにエラーが発生しました.
RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM

解決:
最初は、ビデオカードのキャッシュが多すぎて、キャッシュをクリーンアップして再起動する必要があるという問題の説明をネット上で探しました.しかし、彼らが紹介した手順に従って、問題は解決されていないことがわかり、グラフィックスカードの状態を確認し、メモリの状態は良好です.最後にstackoverflowで見つけた解決方法
方法:
もし状況が私と同じなら、私たちが入力したデータはGPUに置かれているはずです.私たちは例えば削除することができます.Cuda()の操作は、CPUにデータを戻してコードを再実行し、より明確なエラー記述を見ることができますが、根本的な問題は私たちがLongタイプのデータを送ったことです(具体的には違いがあるかもしれませんが)、モデルが望んでいるのはFloatタイプのデータです:解決:
data = data.float().cuda()

具体的な解決には違いがあるかもしれないが、考え方はそうであるべきだ.
END~