RuntimeError: arguments are located on different GPUs at /opt/conda/conda-bld/pytorch_1532581333611/work/aten/src/THC/generated/../generic/THCTensorMathPointwise.cu:314

louis-she / sfd.pytorch

S3FD: single shot face detector in pytorch

MIT License

113 stars 20 forks source link

RuntimeError: arguments are located on different GPUs at /opt/conda/conda-bld/pytorch_1532581333611/work/aten/src/THC/generated/../generic/THCTensorMathPointwise.cu:314 #31

Closed lilmangolil closed 5 years ago

lilmangolil commented 6 years ago

I want to know how to deal with this kind of error.Thank you very much! the error is: RuntimeError: arguments are located on different GPUs at /opt/conda/conda-bld/pytorch_1532581333611/work/aten/src/THC/generated/../generic/THCTensorMathPointwise.cu:314

louis-she commented 5 years ago

Did you modify the device option in config.py ?