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);
}
