luanfujun / deep-painterly-harmonization

Code and data for paper "Deep Painterly Harmonization": https://arxiv.org/abs/1804.03189
6.07k stars 626 forks source link

CUDA #37

Open ganggangzhou opened 5 years ago

ganggangzhou commented 5 years ago

I tried to write the part of matlab dealing with mask in gen_all.py, but I got an error:

/home/wyy/torch/install/bin/luajit: neural_gram.lua:54: cuda runtime error (10) : invalid device ordinal at /home/wyy/torch/extra/cutorch/init.c:734 stack traceback: [C]: in function 'setDevice' neural_gram.lua:54: in function 'main' neural_gram.lua:450: in main chunk [C]: in function 'dofile' .../wyy/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk [C]: at 0x00405d50

can you give me some advice? Thank you very much!!!!!