RuntimeError: arguments are located on different GPUs at /opt/conda/conda-bld/pytorch_1533672544752/work/aten/src/THC/generated/../generic/THCTensorMathPointwise.cu:314
Closed this issue · 1 comments
WesLee88524 commented
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
ok,I work out.
"import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0""