怎么设置只在cpu训练