Rendering: cpu vs gpu

Open ivanpuhachov opened this issue 4 years ago • 2 comments

In your generative_models/rendering.py you have force_cpu=True, but it works fine with all tensors on GPU (after little tweaks like cuda() insted of cpu() here and there). However, it does not give any performance boost (mnist epoch still takes 15 minutes on my machine). Why is that? Am I missing something?

Cheers, Ivan

Apr 22 '21 22:04 ivanpuhachov

I would also be interested in the GPU vs CPU discussion. Installing with CUDA and PyTorch did not really change performance for me. It does not even show the GPU being used in the activity monitor, but PyTorch insists that the tensors are on the GPU.

Dec 05 '21 11:12 flono