Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Can I run this code on non-GPU machine #13

Open
Wanghaiyang0211 opened this issue Jan 18, 2019 · 0 comments
Open

Can I run this code on non-GPU machine #13

Wanghaiyang0211 opened this issue Jan 18, 2019 · 0 comments

Comments

@Wanghaiyang0211
Copy link

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

@Wanghaiyang0211 Wanghaiyang0211 changed the title Can I run dispnet on non-GPU machine Can I run this code on non-GPU machine Jan 18, 2019
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant