facebookresearch / theseus

A library for differentiable nonlinear optimization

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

invoking CPU solving

WeiXiCZ opened this issue · comments

commented

❓ Questions and Help

Hi!Thank you for your outstanding work. I am using your solver, but for some reason, I can only invoke CPU solving and cannot use CUDA. Is there any way to prevent the solver from initializing CUDA?

commented

Oh, sorry to bother you. I solved the problem by reinstalling PyTorch cpuonly😅