run code without CUDA

[chiangandy]

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?

[chiangandy]

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.