diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 96c110a..fac41dc 100644 --- a/.pre-commit-config.yaml +++ b/.pre-commit-config.yaml @@ -52,6 +52,7 @@ repos: [.]flake8[.]cython$| meta[.]yaml$| setup[.]cfg$ + - id: verify-alpha-spec - repo: https://github.com/rapidsai/dependency-file-generator rev: v1.15.1 hooks: diff --git a/dependencies.yaml b/dependencies.yaml index 530d295..2a71d15 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -3,6 +3,8 @@ files: all: output: [conda] matrix: + # CUDA versions match PyTorch releases, not RAPIDS + # ref: https://pytorch.org/get-started/locally/ cuda: ["11.8", "12.1", "12.4"] arch: [x86_64] includes: