1

有谁知道代码流如何在 SLURM 中分配 GPU？我已经检查过了，发现 CUDA_VISIBLE_DEVICES 环境变量在代码中的任何地方都没有更新。那么它实际上是如何在代码中完成的呢？

我的目标是使用环境变量进行分配添加新的硬件支持。

4

1 回答 1

1

看看 gpu gres 插件。代码在其中修改了 CUDA_VISIBLE_DEVICES 环境变量。文件路径为：

src/plugins/gres/gpu/gres_gpu.c
于 2015-04-23T09:59:33.130 回答