限制用户访问 GPU

Question

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

提前致谢！

Answer 1

添加群组库达到你的 Ubuntu，然后设置750*/dev/nvidia** 设备的权限，并将组所有权更改为库达.
将那些被授权使用 GPU 的用户添加到组中库达。

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