On Windows 10 with pytorch, it will only use the CPU despite saying it's using the GPU

[Sister-Perish]

Here's the output it gives when run:

MatplotlibDeprecationWarning:
The support for Qt4 was deprecated in Matplotlib 3.3 and will be removed two minor releases later.
from matplotlib.backends.qt_compat import QtGui
[win_size] = 512
[image_file] = test_imgs/mortar_pestle.jpg
[gpu] = 0
[cpu_mode] = False
[color_prototxt] = ./models/reference_model/deploy_nodist.prototxt
[color_caffemodel] = ./models/reference_model/model.caffemodel
[dist_prototxt] = ./models/reference_model/deploy_nopred.prototxt
[dist_caffemodel] = ./models/reference_model/model.caffemodel
[color_model] = ./models/pytorch/caffemodel.pth
[backend] = pytorch
[pytorch_maskcent] = False
[load_size] = 256
ColorizeImageTorch instantiated
path = ./models/pytorch/caffemodel.pth
Model set! dist mode? False
ColorizeImageTorch instantiated
path = ./models/pytorch/caffemodel.pth
Model set! dist mode? True
test_imgs/mortar_pestle.jpg
scale = 2.000000

But the GPU does nothing, while the CPU is maxed out and the colorization is very slow.
I have installed all of the GPU stuff and verified it's working, so I don't see what the problem is.

[semel1]

The same with me. GPU utilization = 0