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

htqin / IR-Net

[CVPR 2020] This project is the PyTorch implementation of our accepted CVPR 2020 paper : forward and backward information retention for accurate binary neural networks.

177 stars 38 forks source link

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

Closed WesLee88524 closed 4 years ago

WesLee88524 commented 4 years ago

A problem will appear when I run "python main.py" in VGG-Small: RuntimeError: arguments are located on different GPUs at /opt/conda/conda-bld/pytorch_1533672544752/work/aten/src/THC/generated/../generic/THCTensorMathPointwise.cu:314 How can I deal with it?

WesLee88524 commented 4 years ago

ok,I work out.

"import os os.environ["CUDA_VISIBLE_DEVICES"] = "0""