Run without GPU - any replacement for nvdiffrast?

sicxu / Deep3DFaceRecon_pytorch

Accurate 3D Face Reconstruction with Weakly-Supervised Learning: From Single Image to Image Set (CVPRW 2019). A PyTorch implementation.

MIT License

1.66k stars 308 forks source link

Run without GPU - any replacement for nvdiffrast? #143

Open luigi-bar opened 1 year ago

luigi-bar commented 1 year ago

Hi, i would like to run this very nice model in a GPU-free environment ( AWS lambda or similar ). After some fiddling with the code , it seems to me that the critical point is the nvdiffrast library that has no CPU support.

Any change to allow to use some alternative? I would be happy to redo the training if needed.

Thanks!!