search

ROCm / ROCm-docker

Dockerfiles for the various software layers defined in the ROCm software platform
MIT License
422 stars 64 forks source link

Separating containers by single GPUs "--gpus" flag does not seem to function #111

Open bennmann opened 1 year ago

bennmann commented 1 year ago

Hello,

The docker run "--gpus" flag fails to limit the number of GPUs in a single machine to a specific container.

If I wanted to run 2 instances of rocm-docker with "--gpus 0" and "--gpus 1", "rocm-smi --showhw" still seems to detect both gpus in all running containers on my system.

Is this expected behavior? Is there another flag that should be used? Should this be a feature request?

Kind regards,

Ben