支持“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。 |
脚本示例
import arcpy
# Set the gpu id to 1
arcpy.env.gpuId = "1"