Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight';

RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 0 does not equal 1 (while checking arguments for cudnn_convolution)
訓練模型的GPU和你指定加載模型地GPU不同,或是多gpu並行模型加載時,容易報此錯誤

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章