gpu [GPU_ID]??? Interactive Deep Colorization windows 10 64-bit

Hello everyone!

I want to use Interactive Deep Colorization but I can’t run the command line below to run the UI

               python ideepcolor.py --gpu [GPU_ID] --backend [CAFFE OR PYTORCH]

I don’t know [GPU_ID] is what.

I use windows 10  64-bit.

I use Pytorch.

How can I find GPU ID?

Thank you for your support!

icolor.png

The --gpu argument will use a value of 0 if not specified. It appears to be a choice between using --gpu or using --cpu_mode. If you have a GPU, use --gpu or --gpu 0. If you don’t have a GPU, use --cpu_mode.