-
Notifications
You must be signed in to change notification settings - Fork 442
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
run code without CUDA #71
Comments
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. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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?
The text was updated successfully, but these errors were encountered: