我有一台装有多个 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 的进一步说明: