5. CPU/GPU模式和设置说明¶
该平台会自动检测运行的环境是否存在GPU,默认需要安装cuda的环境中才可以使用GPU。
5.1. 使用CPU运行程序¶
无需设置,可自动识别
5.2. 使用单一/多GPU运行程序¶
用户需要在集成调度接口~/test/testimport.py
中,对参数--GPU_Config
进行设置
parser.add_argument(
'--GPU_Config',
type=str,
# 数目,index设置
default=["2","0,1"])
说明:
默认输入的是GPU数目,第二个是GPU的编号,程序会根据用户输入的信息自动校验实际运行环境的GPU情况,合理做出判断,并指定一个可用的GPU设备号作为运行用的设备