search

ProGamerGov / neural-style-pt

PyTorch implementation of neural style transfer algorithm
MIT License
833 stars 178 forks source link

Using without CUDA? #78

Closed bestguy closed 3 years ago

bestguy commented 3 years ago

It's been a while but used to be able to run https://github.com/jcjohnson/neural-style without CUDA support (Macbook Pro)

Is this project able to do same? I see AssertionError: Torch not compiled with CUDA enabled after successful install and unsure if required or how to disable.

ProGamerGov commented 3 years ago

@bestguy Have you tried using -gpu c yet? By default neural-style-pt tries to use your first GPU with -gpu 0.

bestguy commented 3 years ago

Ah thanks! Totally forgot, I appreciate the help and this library. Much easier than the original neural-style!