Specify Gpu device #260
-
|
Dear all, I am using pyamrex with cuda backend. However, I cannot find where I can specify the gpu device_id. Many thanks for your feedback. Best |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 1 reply
-
|
For single process runs, we always use the first visible device. You can use CUDA_VISIBLE_DEVICES to control which devices are visible. https://developer.nvidia.com/blog/cuda-pro-tip-control-gpu-visibility-cuda_visible_devices/ |
Beta Was this translation helpful? Give feedback.
-
|
Agreed. Alternatively, you can switch the device context with, e.g., cupy, and we will use it. This this would be a bit advanced, because you want to do this before we do any device memory allocations in AMReX. |
Beta Was this translation helpful? Give feedback.
For single process runs, we always use the first visible device. You can use CUDA_VISIBLE_DEVICES to control which devices are visible.
https://developer.nvidia.com/blog/cuda-pro-tip-control-gpu-visibility-cuda_visible_devices/