Deep3DFaceRecon_pytorch icon indicating copy to clipboard operation
Deep3DFaceRecon_pytorch copied to clipboard

How to run this on CPU device

Open rayguang opened this issue 2 years ago • 1 comments

I am running a Macbook M1 and it seems to take quite some work to work with the device mps.

I am checking if there is an easy way to just run the code on cpu device. I have tried to set --gpu_ids to -1 when running for example test.py but no luck,

I used the following versions of torch and python 3.9 torch 1.13.1 via pypi

rayguang avatar Feb 03 '23 02:02 rayguang

I have the same problem.

I tried to set "mps" instead of "cuda" in test.py - line 39 device = torch.device("mps" if torch.backends.mps.is_available() else rank)

However, several erros still occur, mainly in Nvdiffrast library.

Ryukhaan avatar Mar 07 '23 13:03 Ryukhaan