private static SessionOptions MakeSessionOptionWithMklDnnProvider() { SessionOptions options = new SessionOptions(); options.AppendExecutionProvider(MklDnnExecutionProviderFactory.Default); options.AppendExecutionProvider(CpuExecutionProviderFactory.Default); return(options); }
/// <summary> /// A helper method to constuct a SessionOptions object for CUDA execution /// </summary> /// <param name="deviceId"></param> /// <returns>A SessionsOptions() object configured for execution on deviceId</returns> public static SessionOptions MakeSessionOptionWithCudaProvider(int deviceId = 0) { CheckCudaExecutionProviderDLLs(); SessionOptions options = new SessionOptions(); if (deviceId == 0) //default value { options.AppendExecutionProvider(CudaExecutionProviderFactory.Default); } else { options.AppendExecutionProvider(new CudaExecutionProviderFactory(deviceId)); } options.AppendExecutionProvider(MklDnnExecutionProviderFactory.Default); options.AppendExecutionProvider(CpuExecutionProviderFactory.Default); return(options); }