Can I run this code on non-GPU machine
Wanghaiyang0211 opened this issue · 0 comments
Wanghaiyang0211 commented
When I compile lmbspecialops with BUILD_WITH_CUDA=OFF, I run the controller.py of css in dispnet3 and I get the following error:
InvalidArgumentError (see above for traceback): No OpKernel was registered to support Op 'Correlation1D' with these attrs. Registered devices: [CPU], Registered kernels:
[[Node: net1/upper/Correlation1D = Correlation1D[T=DT_FLOAT, corr_type="mult", do_abs=false, kernel_size=1, max_displacement=40, pad_size=40, single_dir=0, stride1=1, stride2=1] (net1/features/conv3/LeakyRelu, net1/features/conv3_2/LeakyRelu)]]
When I compile lmbspecialops with BUILD_WITH_CUDA=on, there is no problem
So I want to know if this code can run on a machine without a GPU.
Thank you