run code without CUDA - Githubissues

williamyang1991 / VToonify

[SIGGRAPH Asia 2022] VToonify: Controllable High-Resolution Portrait Video Style Transfer

Other

3.53k stars 442 forks source link

Open chiangandy opened 7 months ago

chiangandy commented 7 months ago

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 commented 7 months ago

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.