限制特定用户访问 Nvidia GPU

限制特定用户访问 Nvidia GPU

我有一台装有多个 GPU 的机器,运行着 ubuntu 16.04。该机器将有多个用户运行不同的实验,我需要限制对 GPU 的访问,以便只有特定用户才能访问特定 GPU。正如限制用户访问 GPU我尝试过更改文件的组和权限/dev/nvidia*,但是,当打开新会话时,它们会恢复到原始设置。我还创建了/etc/modprode.d/nvidia.conf包含options nvidia NVreg_DeviceFileGID=120,但这也不起作用。是我放nvidia.conf错了地方还是语法错误?如果有任何其他建议,我将不胜感激!

更新:

正如@TadejP 建议的那样,我已进行编辑etc/rc.local以包括:

chmod 666 /dev/nvidia5
chgrp nvidia_5 /dev/nvidia5
/bin/usr/nvidia-persistenced --persistence-mode --user <USERNAME> --verbose

但是,这不管用。我一启动,nvidia-smi设备文件就恢复到以前的状态。

答案1

其他指令对我来说也不起作用,所以我们可以在重启后输入命令来执行操作。例如,要限制仅第二张卡,请在 /etc/rc.local 中输入命令:

chmod 660 /dev/nvidia1
chgrp cuda /dev/nvidia1

不在该组中的其他用户cuda甚至看不到带有 的设备nvidia-smi。有关使用 /etc/rc.local 的进一步说明:

相关内容