解决:将送入模型的batch_size调整小点,已超出GPU的处理范围
RuntimeError: NVML_SUCCESS == DriverAPI::get()->nvmlDeviceGetHandleByPciBusId_v2_( pci_id, &nvml_dev
最新推荐文章于 2024-06-13 15:28:12 发布
解决:将送入模型的batch_size调整小点,已超出GPU的处理范围