将 GPU 资源限制给特定进程

将 GPU 资源限制给特定进程

ulimit命令可以限制Linux机器中进程的内存。同样,Linux 中是否有任何命令可以运行 GPU 资源有限的进程。例如,我有一台 64 GB GPU 机器,出于某种实验原因,我想限制某个进程(YOLO)仅使用 1 GB GPU。

相关内容