Bundle libcupti with libtorch to make it work with cuda-platform-redist

bytedeco · Jan 8, 2023 · 1d68090 · 1d68090
1 parent 1486766
commit 1d68090
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java b/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java
@@ -1735,7 +1735,8 @@
         ),
         @Platform(
             value = {"linux", "macosx", "windows"},
-            link = {"c10", "c10_cuda", "torch_cpu", "torch_cuda", "torch"},
+            linkpath = {"/usr/local/cuda/lib64/", "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.8/lib/x64/"},
+            link = {"c10", "c10_cuda", "torch_cpu", "torch_cuda", "torch", "cupti@11.8"},
             preloadpath = "C:/Program Files/NVIDIA Corporation/NvToolsExt/bin/x64/",
             extension = "-gpu"
         ),