我有一个具有6个v100的机器学习设备,人们可以通过Docker容器内的Jupyter访问它们。我想控制谁使用哪个图形卡,所以我必须设置一个环境变量。
因此,我想将环境变量NVIDIA_VISIBLE_DEVICES
设置为特定的图形卡。但是在Jupyter(在docker容器上运行)中,用户可以编辑此变量,因此每个人都可以访问每个GPU。
如何使该变量不可写?
答案 0 :(得分:2)
将readonly NVIDIA_VISIBLE_DEVICES
添加到linux的某个起始文件(也许是.bashrc)中,在启动容器时使用-e
设置NVIDIA_VISIBLE_DEVICES
的默认值。
然后用户无法在jupyter中进行更改。