我有一个带有 6 个 v100 的机器学习设备,人们可以通过 Docker 容器内的 Jupyter 访问它们。我想控制谁使用哪个显卡,所以我必须设置一个环境变量。
因此,我想将环境变量 NVIDIA_VISIBLE_DEVICES
设置为特定的显卡。但在 Jupyter(在 docker 容器上运行)中,用户可以编辑此变量,这样每个人都可以访问每个 GPU。
如何使该变量不可写?
最佳答案
将readonly NVIDIA_VISIBLE_DEVICES
添加到linux的某个启动文件(可能是.bashrc),使用-e
设置NVIDIA_VISIBLE_DEVICES
的默认值启动容器。
然后用户无法在 jupyter 中更改它。
关于Docker使环境变量不可写,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51240851/