Pytork如何设置每个进程的GPU限制?

2024-04-25 11:44:08 发布

您现在位置:Python中文网/ 问答频道 /正文

Pytorch如何设置GPU内存限制

当我启动uwsgi并设置2名工人时。然后我的GPU内存将共享两个人的所有内存

例如4GB GPU->;。第一个工人使用:2 GB。第二工人2GB

如何设置第一个工作进程仅使用1GB第二个工作进程使用1GB


Tags: 内存gtgpu进程pytorchuwsgi工人gb