diff --git a/Artifacts.toml b/Artifacts.toml index fb605d7..e620db0 100644 --- a/Artifacts.toml +++ b/Artifacts.toml @@ -1,57 +1,57 @@ [[onnxruntime_cpu]] arch = "x86_64" -git-tree-sha1 = "ee0b3b47cb5c295d8e2df9b704d472fa827fbb59" +git-tree-sha1 = "03ca4920f8c66efacaa58cdb30ee6da83b7a40e0" lazy = true os = "windows" [[onnxruntime_cpu.download]] - sha256 = "f132ee57ab970c496f08e344bef60f03d3cf61475c93628feef2a1ac94e88aa1" - url = "https://github.com/jw3126/ONNXRunTimeArtifacts/releases/download/v1.15.1-rc1/onnxruntime-win-x64-1.15.1.tgz" + sha256 = "0dbe0bb59e2c000d9b331d1dfd2a61a7af2cfea1d5391d369fcd8d3ea504b3b8" + url = "https://github.com/jw3126/ONNXRunTimeArtifacts/releases/download/v1.20.1-rc1/onnxruntime-win-x64-1.20.1.tgz" [[onnxruntime_cpu]] arch = "x86_64" -git-tree-sha1 = "fe904e447d592a9cf7aaac236a58a94f91f761d1" +git-tree-sha1 = "7a5a70a99ed5931ff451b6e70a29944cd3d4b54d" lazy = true libc = "glibc" os = "linux" [[onnxruntime_cpu.download]] - sha256 = "5492f9065f87538a286fb04c8542e9ff7950abb2ea6f8c24993a940006787d87" - url = "https://github.com/microsoft/onnxruntime/releases/download/v1.15.1/onnxruntime-linux-x64-1.15.1.tgz" + sha256 = "67db4dc1561f1e3fd42e619575c82c601ef89849afc7ea85a003abbac1a1a105" + url = "https://github.com/microsoft/onnxruntime/releases/download/v1.20.1/onnxruntime-linux-x64-1.20.1.tgz" [[onnxruntime_cpu]] arch = "x86_64" -git-tree-sha1 = "fcb94c1a37609b80bf69b143bfcd60b0ff1fbef1" +git-tree-sha1 = "4652621efdd6d8c61987734403a7e7e1fff19079" lazy = true os = "macos" [[onnxruntime_cpu.download]] - sha256 = "ecb7651c216fe6ffaf4c578e135d98341bc5bc944c5dc6b725ef85b0d7747be0" - url = "https://github.com/microsoft/onnxruntime/releases/download/v1.15.1/onnxruntime-osx-universal2-1.15.1.tgz" + sha256 = "da4349e01a7e997f5034563183c7183d069caadc1d95f499b560961787813efd" + url = "https://github.com/microsoft/onnxruntime/releases/download/v1.20.1/onnxruntime-osx-universal2-1.20.1.tgz" [[onnxruntime_cpu]] arch = "aarch64" -git-tree-sha1 = "fcb94c1a37609b80bf69b143bfcd60b0ff1fbef1" +git-tree-sha1 = "9f203a6745ce1e17f0afb8528688b95d1791d851" lazy = true os = "macos" [[onnxruntime_cpu.download]] - sha256 = "ecb7651c216fe6ffaf4c578e135d98341bc5bc944c5dc6b725ef85b0d7747be0" - url = "https://github.com/microsoft/onnxruntime/releases/download/v1.15.1/onnxruntime-osx-universal2-1.15.1.tgz" + sha256 = "b678fc3c2354c771fea4fba420edeccfba205140088334df801e7fc40e83a57a" + url = "https://github.com/microsoft/onnxruntime/releases/download/v1.20.1/onnxruntime-osx-arm64-1.20.1.tgz" [[onnxruntime_gpu]] arch = "x86_64" -git-tree-sha1 = "68acc458450c6cff16f55c8cedba213f1eeeb678" +git-tree-sha1 = "9740e9c731f3aeaa8ea93e25a787e91cf8f64bd7" lazy = true os = "windows" [[onnxruntime_gpu.download]] - sha256 = "94aea5b7ec2ffeecca8225816116922cab7ce0f4f2a5516c28e241cdbf4c04db" - url = "https://github.com/jw3126/ONNXRunTimeArtifacts/releases/download/v1.15.1-rc1/onnxruntime-win-x64-gpu-1.15.1.tgz" + sha256 = "622828adb36268ce58d69eade91827d92354388f6e6a3be777fffe54acab84a2" + url = "https://github.com/jw3126/ONNXRunTimeArtifacts/releases/download/v1.20.1-rc1/onnxruntime-win-x64-gpu-1.20.1.tgz" [[onnxruntime_gpu]] arch = "x86_64" -git-tree-sha1 = "d7e3cdf6598724d24e8f854817c6e71a1f6fe57e" +git-tree-sha1 = "478f998b8d737218ef0cc06d26e91288d1e72f3b" lazy = true libc = "glibc" os = "linux" [[onnxruntime_gpu.download]] - sha256 = "eab891393025edd5818d1aa26a42860e5739fcc49e3ca3f876110ec8736fe7f1" - url = "https://github.com/microsoft/onnxruntime/releases/download/v1.15.1/onnxruntime-linux-x64-gpu-1.15.1.tgz" + sha256 = "6bfb87c6ebe55367a94509b8ef062239e188dccf8d5caac8d6909b2344893bf0" + url = "https://github.com/microsoft/onnxruntime/releases/download/v1.20.1/onnxruntime-linux-x64-gpu-1.20.1.tgz" diff --git a/Project.toml b/Project.toml index ea87d48..193be05 100644 --- a/Project.toml +++ b/Project.toml @@ -1,7 +1,7 @@ name = "ONNXRunTime" uuid = "e034b28e-924e-41b2-b98f-d2bbeb830c6a" authors = ["Jan Weidner and contributors"] -version = "1.2.0" +version = "1.3.0" [deps] ArgCheck = "dce04be8-c92d-5529-be00-80e4d2c0e197" @@ -18,7 +18,7 @@ CEnum = "0.4, 0.5" CUDA = "4, 5" DataStructures = "0.18" DocStringExtensions = "0.8, 0.9" -cuDNN = "~1.1, ~1.2, =1.3.0" +cuDNN = "~1.3, ~1.4" julia = "1.9" [extensions] diff --git a/README.md b/README.md index 2ffb8bf..bd41c25 100644 --- a/README.md +++ b/README.md @@ -30,7 +30,7 @@ Dict{String, Matrix{Float32}} with 1 entry: ``` For GPU usage the CUDA and cuDNN packages are required and the CUDA -runtime needs to be set to 11.8 or a later 11.x version. To set this +runtime needs to be set to 12.0 or a later 12.x version. To set this up, do ```julia @@ -38,7 +38,7 @@ pkg> add CUDA cuDNN julia> import CUDA -julia> CUDA.set_runtime_version!(v"11.8") +julia> CUDA.set_runtime_version!(v"12.0") ``` Then GPU inference is simply @@ -109,14 +109,14 @@ and import CUDA and cuDNN. Additionally a supported CUDA runtime version needs to be used, which can be somewhat tricky to set up for the tests. -First some background. What `CUDA.set_runtime_version!(v"11.8")` +First some background. What `CUDA.set_runtime_version!(v"12.0")` effectively does is to 1. Add a `LocalPreferences.toml` file containing ``` [CUDA_Runtime_jll] -version = "11.8" +version = "12.0" ``` 2. In `Project.toml`, add diff --git a/src/versions.jl b/src/versions.jl index 52c9a87..2e87e1c 100644 --- a/src/versions.jl +++ b/src/versions.jl @@ -16,6 +16,6 @@ # is *not* accepted. Presumably CUDA runtime follows semantic # versioning so this can automatically be set to the next major # version. -const onnxruntime_version = v"1.15.1" -const cuda_runtime_supported_version = v"11.8" +const onnxruntime_version = v"1.20.1" +const cuda_runtime_supported_version = v"12.0" const cuda_runtime_upper_bound = VersionNumber(cuda_runtime_supported_version.major + 1) diff --git a/test/LocalPreferences.toml b/test/LocalPreferences.toml index 5da06c7..ef09e81 100644 --- a/test/LocalPreferences.toml +++ b/test/LocalPreferences.toml @@ -1,2 +1,2 @@ [CUDA_Runtime_jll] -version = "11.8" +version = "12.0" diff --git a/test/Project.toml b/test/Project.toml index 2fe8444..fe5017c 100644 --- a/test/Project.toml +++ b/test/Project.toml @@ -5,7 +5,7 @@ cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd" [compat] CUDA = "5" -cuDNN = "1.2" +cuDNN = "1.3" [extras] CUDA_Runtime_jll = "76a88914-d11a-5bdc-97e0-2f5a05c973a2"