限制用户访问 GPU

Question

我希望限制某些用户在联合使用的工作站上使用 GPU（在保留内存方面）。因此，如果用户尝试通过 SSH 访问 GPU，例如通过 CUDA，该请求将被拒绝。任何想法从哪里开始寻找？

提前致谢！

Answer 1

将组cuda添加到您的 Ubuntu，然后在 */dev/nvidia** 设备上设置750权限并将组所有权更改为cuda。
将那些被授予使用 GPU 的用户添加到cuda组中。

还有一些驱动程序参数可以在加载 nvidia 驱动程序期间为您执行此操作。（NVreg_DeviceFileGID和NVreg_DeviceFileMode）。