saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-17 21:10:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Revert to cuda 9.1 for package release (#546)

Browse source
This commit is contained in:
jignparm 2019-03-05 17:20:51 -08:00 committed by GitHub
parent 47a9abd212
commit 94bd74190a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
csharp/src/Microsoft.ML.OnnxRuntime/SessionOptions.cs
View file

@ -25,7 +25,7 @@ namespace Microsoft.ML.OnnxRuntime
{
public IntPtr _nativePtr;
protected static readonly Lazy<SessionOptions> _default = new Lazy<SessionOptions>(MakeSessionOptionWithCpuProvider);
private static string[] cudaDelayLoadedLibs = { "cublas64_100.dll", "cudnn64_7.dll" };
private static string[] cudaDelayLoadedLibs = { "cublas64_91.dll", "cudnn64_7.dll" };
/// <summary>
/// Constructs an empty SessionOptions
@ -97,7 +97,7 @@ namespace Microsoft.ML.OnnxRuntime
GetSystemDirectory(sysdir, (uint)sysdir.Capacity);
throw new OnnxRuntimeException(
ErrorCode.NoSuchFile,
$"kernel32.LoadLibrary():'{dll}' not found. CUDA 10.0 is required for GPU execution. " +
$"kernel32.LoadLibrary():'{dll}' not found. CUDA is required for GPU execution. " +
$". Verify it is available in the system directory={sysdir}. Else copy it to the output folder."
);
}