question about multi-core and multi GPU

mitsuba-renderer / enoki

Enoki: structured vectorization and differentiation on modern processor architectures

Other

1.26k stars 94 forks source link

Closed lvjiahui closed 4 years ago

lvjiahui commented 4 years ago

I am importing the Enoki in python and using dynamic array. I find it only uses one core of cpu. Should I use multithread myself or Enoki surpport multi-core parallal?
I have multi GPU, can I specify which GPU Enoki uses?

Thanks for your help!

Speierers commented 4 years ago

Dynamic arrays in Enoki do not take advantage of multiple cores on the CPU. For this you will have to parallelize your code manually.

I believe you should be able to select a GPU with the environement variable CUDA_VISIBLE_DEVICES=GPU_ID. Let me know if this works for you.

lvjiahui commented 4 years ago

CUDA_VISIBLE_DEVICES=GPU_ID python

It works. thanks!