How can it run on GPU?

[Jiajia0319]

When I use the pretrained model, the result is really good. But it costs 45 min to make a image (3450x2230) to 2x. So should I try it with GPU? Because it runs on CPU default. Thanks a lot!

[tsurumeso]

You can use GPU specifying a GPU ID with --gpu or -g option. Sorry for not explaining enough on README.

# Run on CPU
python waifu2x.py -m scale -i path/to/image/or/directory -a 0

# Run on GPU
python waifu2x.py -m scale -i path/to/image/or/directory -a 0 -g 0

[Jiajia0319]

You can use GPU specifying a GPU ID with --gpu or -g option. Sorry for not explaining enough on README.

# Run on CPU
python waifu2x.py -m scale -i path/to/image/or/directory -a 0

# Run on GPU
python waifu2x.py -m scale -i path/to/image/or/directory -a 0 -g 0

Thanks a lot! I have resolved that.