run code without CUDA
chiangandy opened this issue · comments
I found style_transfer.py has option --cpu which can be run on non-cuda environment. But when I try this..
python3 style_transfer.py --cpu --scale_image --content a1.png --ckpt ./checkpoint/vtoonify.pt
It response ....
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
how to solve this?
answer by my self, we can modify 3 files and put import library to use _cpu then command line need to set --cpu then everything will be perfact.