luanfujun / deep-painterly-harmonization

Code and data for paper "Deep Painterly Harmonization": https://arxiv.org/abs/1804.03189

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CUDA

ganggangzhou opened this issue · comments

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!!!!!