Skip to content

Does it support running on cpu?  #310

Closed
@margaretmm

Description

@margaretmm

I ran this command: python2 test.py --dataroot datasets/horse2zebra/testA --name horse2zebra_pretrained --model test

and it return :

THCudaCheck FAIL file=torch/csrc/cuda/Module.cpp line=32 error=35 : CUDA driver version is insufficient for CUDA runtime version
Traceback (most recent call last):
File "test.py", line 10, in
opt = TestOptions().parse()
File "/home/AI/pytorch-CycleGAN-and-pix2pix/options/base_options.py", line 115, in parse
torch.cuda.set_device(opt.gpu_ids[0])
File "/usr/lib64/python2.7/site-packages/torch/cuda/init.py", line 262, in set_device
torch._C._cuda_setDevice(device)
RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version at torch/csrc/cuda/Module.cpp:32

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions