Gpu allocation implementation

Hello! I wonder how ray implement gpu assignment with a specific allocation, does it do it by setting env virable CUDA_VISIBLE_DEVICES? please tell me where the code is.

yes, it’s implemented via CUDA_VISIBLE_DEVICES. GPU Support — Ray 3.0.0.dev0 has some details on Ray’s GPU support.

The related code are here: link.