Honggu Liu

icon company University of Science and Technology of China icon location Heifei, Anhui Province, China icon location Ph.D. student of AI security.

Results 1 issues of Honggu Liu

RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 1 does not equal 0 (while checking arguments for cudnn_convolution)

2 comment

Hello, When I used multiple gpus to train model, I usually get this error. `RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument...