指定启动时使用的 GPU

指定启动时使用的 GPU

我目前正在使用 Azure GPU VM(使用 M60 Nvidia 显卡的 NV6)。我在此 VM 上执行基准测试,目前没有任何问题。现在,我在具有 2 个 GPU(或至少 Windows 服务器将其视为 2 个 GPU)的 NV12 上执行相同的基准测试。

我想启动我的应用程序两次,每个应用程序都有自己的 GPU。我有点迷茫,不知道在哪里可以为程序指定 GPU。是 Windows 配置/poweshell 脚本,还是要设置的 Nvidia Grids 配置,或者我必须在我的应用程序中添加一些代码来选择要使用的 GPU?

谢谢你,

相关内容