diff --git a/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java b/pytorch/src/main/java/org/bytedeco/pytorch/presets/torch.java index 18d481d541e..f4be35540e0 100644 --- 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 = {"gomp@.1", "iomp5", "omp", "tbb@.2", "asmjit", "fbgemm", "cupti@.12"}, - 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 index 223849e4d77..ad06117d870 100644 --- 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/" }