Update CUDA library paths to 12.3

bytedeco · Nov 3, 2023 · fe140fd · fe140fd
1 parent 95496c6
commit fe140fd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java b/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java
@@ -79,15 +79,15 @@
             value = {"linux", "macosx", "windows"},
             link = { "c10", "c10_cuda", "torch_cpu", "torch_cuda", "torch" },
             preload = {"[email protected]", "iomp5", "omp", "[email protected]", "asmjit", "fbgemm", "[email protected]"},
-            includepath = {"/usr/local/cuda/include", "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.1/include/"},
+            includepath = {"/usr/local/cuda/include", "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.3/include/"},
             preloadpath = {
-                "/usr/local/cuda-12.1/lib64/",
-                "/usr/local/cuda-12.1/extras/CUPTI/lib64/",
+                "/usr/local/cuda-12.3/lib64/",
+                "/usr/local/cuda-12.3/extras/CUPTI/lib64/",
                 "/usr/local/cuda/lib64/",
                 "/usr/local/cuda/extras/CUPTI/lib64/",
                 "/usr/lib64/",
-                "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.1/lib/x64/",
-                "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.1/extras/CUPTI/lib64/",
+                "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.3/lib/x64/",
+                "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.3/extras/CUPTI/lib64/",
                 "C:/Program Files/NVIDIA Corporation/NvToolsExt/bin/x64/",
             },
 

diff --git a/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch_cuda.java b/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch_cuda.java
@@ -46,8 +46,8 @@
             },
             link = { "cudart", "cusparse" },
             linkpath = {
-                "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.1/lib/x64/",
-                "/usr/local/cuda-12.1/lib64/",
+                "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.3/lib/x64/",
+                "/usr/local/cuda-12.3/lib64/",
                 "/usr/local/cuda/lib64/",
                 "/usr/lib64/"
             }