Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update onnxruntime to 1.20.1 #40

Merged
merged 9 commits into from
Dec 1, 2024
Merged
Show file tree
Hide file tree
Changes from 8 commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/CI.yml
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ jobs:
fail-fast: false
matrix:
version:
- '1.9'
- '1.10'
- '1'
- 'nightly'
os:
Expand Down
36 changes: 18 additions & 18 deletions Artifacts.toml
Original file line number Diff line number Diff line change
@@ -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"
6 changes: 3 additions & 3 deletions Project.toml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name = "ONNXRunTime"
uuid = "e034b28e-924e-41b2-b98f-d2bbeb830c6a"
authors = ["Jan Weidner <[email protected]> and contributors"]
version = "1.2.0"
version = "1.3.0"

[deps]
ArgCheck = "dce04be8-c92d-5529-be00-80e4d2c0e197"
Expand All @@ -18,8 +18,8 @@ CEnum = "0.4, 0.5"
CUDA = "4, 5"
DataStructures = "0.18"
DocStringExtensions = "0.8, 0.9"
cuDNN = "~1.1, ~1.2, =1.3.0"
julia = "1.9"
cuDNN = "~1.3, ~1.4"
julia = "1.10"

[extensions]
CUDAExt = ["CUDA", "cuDNN"]
Expand Down
8 changes: 4 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -30,15 +30,15 @@ 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
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
Expand Down Expand Up @@ -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
Expand Down
4 changes: 2 additions & 2 deletions src/versions.jl
Original file line number Diff line number Diff line change
Expand Up @@ -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)
2 changes: 1 addition & 1 deletion test/LocalPreferences.toml
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
[CUDA_Runtime_jll]
version = "11.8"
version = "12.0"
2 changes: 1 addition & 1 deletion test/Project.toml
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"

[compat]
CUDA = "5"
cuDNN = "1.2"
cuDNN = "1.4"
jw3126 marked this conversation as resolved.
Show resolved Hide resolved

[extras]
CUDA_Runtime_jll = "76a88914-d11a-5bdc-97e0-2f5a05c973a2"
Loading