update torch getting started to workaround dotnet/fsharp#10136

dsyme · Sep 16, 2020 · 470ba1d · 470ba1d
1 parent 8494063
commit 470ba1d
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/docs/getting-started-torch.fsx b/docs/getting-started-torch.fsx
@@ -9,7 +9,21 @@
 (*** condition: ipynb ***)
 #if IPYNB
 #r "nuget: DiffSharp-cpu,{{fsdocs-package-version}}"
+#endif // IPYNB
+
+(*** condition: fsx ***)
+#if FSX
+// This is a workaround for https://github.com/dotnet/fsharp/issues/10136, necessary in F# scripts and .NET Interactive
+System.Runtime.InteropServices.NativeLibrary.Load(let path1 = System.IO.Path.GetDirectoryName(typeof<DiffSharp.dsharp>.Assembly.Location) in if System.Runtime.InteropServices.RuntimeInformation.IsOSPlatform(System.Runtime.InteropServices.OSPlatform.Linux) then path1 + "/../../../../libtorch-cpu/1.5.6/runtimes/linux-x64/native/libtorch.so" else path1 + "/../../../../libtorch-cpu/1.5.6/runtimes/win-x64/native/torch_cpu.dll")
+#r "nuget: DiffSharp-cpu,{{fsdocs-package-version}}"
+#endif // FSX
+
+(*** condition: ipynb ***)
+#if IPYNB
+// This is a workaround for https://github.com/dotnet/fsharp/issues/10136, necessary in F# scripts and .NET Interactive
+System.Runtime.InteropServices.NativeLibrary.Load(let path1 = System.IO.Path.GetDirectoryName(typeof<DiffSharp.dsharp>.Assembly.Location) in if System.Runtime.InteropServices.RuntimeInformation.IsOSPlatform(System.Runtime.InteropServices.OSPlatform.Linux) then path1 + "/../../../../libtorch-cpu/1.5.6/runtimes/linux-x64/native/libtorch.so" else path1 + "/../../../../libtorch-cpu/1.5.6/runtimes/win-x64/native/torch_cpu.dll")
 
+// Set up formatting for notebooks
 Formatter.SetPreferredMimeTypeFor(typeof<obj>, "text/plain")
 Formatter.Register(fun (x:obj) (writer: TextWriter) -> fprintfn writer "%120A" x )
 #endif // IPYNB