GPU ID(环境设置)

支持“GPU ID”环境的工具可标识用于处理数据的 GPU。

用法说明

  • 在有多个可用计算 GPU 的情况下,确定使用哪个 GPU。
  • 使用 CUDA 附带的 NVIDIA 系统管理界面 (nvidia-smi) 命令工具来检查 GPU 的顺序和状态。 Nvidia-smi 命令工具还可提供诸如 GPU 负载、内存负载、温度和其他 GPU 性能特征等信息。
  • 建议仅将一个 GPU 用于显示(通过将显示器连接到该 GPU),并将另一个 GPU 设置为仅用于计算。 由此将确保计算和显示 GPU 不会相互干扰,并将最大限度地提高效率。

对话框语法

设置用于处理数据的 GPU 标识号。

  • GPU ID - 用于处理数据的 GPU 标识号。默认值为 0。

脚本语法

arcpy.env.gpuId = GPU_ID

参数说明

gpuId

用于处理数据的 GPU 标识号。这是一个整数,且默认值为 0。

gpuId 语法

脚本示例

import arcpy
# Set the gpu id to 1
arcpy.env.gpuId = "1"