Skip to content

Commit

Permalink
updated CUDNN
Browse files Browse the repository at this point in the history
  • Loading branch information
cocoa-xu committed Feb 13, 2024
1 parent ef94348 commit 58e5713
Show file tree
Hide file tree
Showing 2 changed files with 25 additions and 23 deletions.
41 changes: 22 additions & 19 deletions .github/workflows/windows-precompile-cuda.yml
Original file line number Diff line number Diff line change
Expand Up @@ -29,28 +29,24 @@ jobs:
fail-fast: false
matrix:
include:
- cuda_ver: "11.1.1"
cuda_id: "111"
OTP_VERSION: "25"
NIF_VERSION: "2.16"
- cuda_ver: "11.4.3"
cuda_id: "114"
OTP_VERSION: "25"
NIF_VERSION: "2.16"
- cuda_ver: "11.8.0"
cuda_id: "118"
cudnn: "8.7.0"
OTP_VERSION: "25"
NIF_VERSION: "2.16"
- cuda_ver: "11.1.1"
cuda_id: "111"
OTP_VERSION: "26"
NIF_VERSION: "2.17"
- cuda_ver: "11.4.3"
cuda_id: "114"
OTP_VERSION: "26"
NIF_VERSION: "2.17"
- cuda_ver: "11.8.0"
cuda_id: "118"
cudnn: "8.7.0"
OTP_VERSION: "26"
NIF_VERSION: "2.17"
- cuda_ver: "12.1.0"
cuda_id: "121"
cudnn: "8.9.0"
OTP_VERSION: "25"
NIF_VERSION: "2.16"
- cuda_ver: "12.1.0"
cuda_id: "121"
cudnn: "8.9.0"
OTP_VERSION: "26"
NIF_VERSION: "2.17"

Expand All @@ -75,12 +71,19 @@ jobs:
arch: x64
vsversion: "2019"

- uses: cocoa-xu/cuda-toolkit@v0.3.0
id: cuda-toolkit
- uses: cocoa-xu/cuda-toolkit@v0.4.0
if: ${{ matrix.cudnn }} == "8.7.0"
with:
cuda: ${{ matrix.cuda_ver }}
cudnn: "8.7.0"
cudnn: ${{ matrix.cudnn }}
cudnn_url: ${{ secrets.CUDA_V870_WINDOWS_X86_64_CUDA11 }}

- uses: cocoa-xu/[email protected]
if: ${{ matrix.cudnn }} == "8.9.0"
with:
cuda: ${{ matrix.cuda_ver }}
cudnn: ${{ matrix.cudnn }}
cudnn_url: ${{ secrets.CUDA_V890_WINDOWS_X86_64_CUDA12 }}

- uses: actions/setup-python@v4
with:
Expand Down
7 changes: 3 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -456,9 +456,8 @@ export EVISION_ENABLE_CUDA=true
## set a CUDA version that matches your local CUDA driver
## (this environment variable is only required for users who'd like to use precompiled binaries)
## available ones are
## 111, for CUDA 11.1.x, built with CUDA 11.1.1
## 114, for CUDA 11.4.x, built with CUDA 11.4.3
## 118, for CUDA 11.8.x, built with CUDA 11.8.0
## 118, for CUDA 11.8+, built with CUDA 11.8.0
## 121, for CUDA 12.1+, built with CUDA 12.1.0
export EVISION_CUDA_VERSION=118
## opencv_contrib modules is enabled by default
export EVISION_ENABLE_CONTRIB=true
Expand Down Expand Up @@ -511,7 +510,7 @@ export EVISION_PRECOMPILED_CACHE_DIR="$(pwd)/.cache"

OpenCV 4.5.5 requires at least CMake 3.5.1.

- Erlang development library/headers. Tested on OTP/25.
- Erlang development library/headers. Tested on OTP/25 and OTP/26.

#### Build from source
To obtain and compile OpenCV's source code from official releases, the following environment variables can affect the build
Expand Down

0 comments on commit 58e5713

Please sign in to comment.