Use CPU to run test.py command get error

[Chen0729]

Hi,

I am trying to run test.py command on CPU so I modified gpu_ids to be 0 with the pre-trained models. I get the following errors:
Traceback (most recent call last):
File "test.py", line 18, in
model = create_model(opt)
File "C:\poseGAN\poseTransfer\Pose-Transfer-master\Pose-Transfer-master\models\models.py", line 13, in create_model
model.initialize(opt)
File "C:\poseGAN\poseTransfer\Pose-Transfer-master\Pose-Transfer-master\models\PATN.py", line 29, in initialize
self.input_P1_set = self.Tensor(nb, opt.P_input_nc, size, size)
File "C:\Users\anaconda3\lib\site-packages\torch\cuda_init_.py", line 172, in _lazy_init
torch._C._cuda_init()
RuntimeError: Found no NVIDIA driver on your system.

May I ask if the software can be run on CPU for the testing phase please ? If yes, how to modify it to make it run on CPU.

Thanks