diff options
author | github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com> | 2022-04-09 12:02:07 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-04-09 12:02:07 +0000 |
commit | f303d4de3bd2245a45b1d33a3fa755bd5bbd015f (patch) | |
tree | 412eeeea02722f1f2d9e14941bf254991013c5fc /doc | |
parent | 01f8e687486c61f5bc2eea142da8cc47369b8fef (diff) | |
parent | 40b03d10a3931055a98981384205bdbc0a8f2bd5 (diff) | |
download | nixpkgs-f303d4de3bd2245a45b1d33a3fa755bd5bbd015f.tar nixpkgs-f303d4de3bd2245a45b1d33a3fa755bd5bbd015f.tar.gz nixpkgs-f303d4de3bd2245a45b1d33a3fa755bd5bbd015f.tar.bz2 nixpkgs-f303d4de3bd2245a45b1d33a3fa755bd5bbd015f.tar.lz nixpkgs-f303d4de3bd2245a45b1d33a3fa755bd5bbd015f.tar.xz nixpkgs-f303d4de3bd2245a45b1d33a3fa755bd5bbd015f.tar.zst nixpkgs-f303d4de3bd2245a45b1d33a3fa755bd5bbd015f.zip |
Merge staging-next into staging
Diffstat (limited to 'doc')
-rw-r--r-- | doc/languages-frameworks/cuda.section.md | 34 | ||||
-rw-r--r-- | doc/languages-frameworks/index.xml | 1 |
2 files changed, 35 insertions, 0 deletions
diff --git a/doc/languages-frameworks/cuda.section.md b/doc/languages-frameworks/cuda.section.md new file mode 100644 index 00000000000..fccf66bf79d --- /dev/null +++ b/doc/languages-frameworks/cuda.section.md @@ -0,0 +1,34 @@ +# CUDA {#cuda} + +CUDA-only packages are stored in the `cudaPackages` packages set. This set +includes the `cudatoolkit`, portions of the toolkit in separate derivations, +`cudnn`, `cutensor` and `nccl`. + +A package set is available for each CUDA version, so for example +`cudaPackages_11_6`. Within each set is a matching version of the above listed +packages. Additionally, other versions of the packages that are packaged and +compatible are available as well. For example, there can be a +`cudaPackages.cudnn_8_3_2` package. + +To use one or more CUDA packages in an expression, give the expression a `cudaPackages` parameter, and in case CUDA is optional +```nix +cudaSupport ? false +cudaPackages ? {} +``` + +When using `callPackage`, you can choose to pass in a different variant, e.g. +when a different version of the toolkit suffices +```nix +mypkg = callPackage { cudaPackages = cudaPackages_11_5; } +``` + +If another version of say `cudnn` or `cutensor` is needed, you can override the +package set to make it the default. This guarantees you get a consistent package +set. +```nix +mypkg = let + cudaPackages = cudaPackages_11_5.overrideScope' (final: prev { + cudnn = prev.cudnn_8_3_2; + }}); +in callPackage { inherit cudaPackages; }; +``` diff --git a/doc/languages-frameworks/index.xml b/doc/languages-frameworks/index.xml index f221693e764..144fa96bac1 100644 --- a/doc/languages-frameworks/index.xml +++ b/doc/languages-frameworks/index.xml @@ -11,6 +11,7 @@ <xi:include href="bower.section.xml" /> <xi:include href="coq.section.xml" /> <xi:include href="crystal.section.xml" /> + <xi:include href="cuda.section.xml" /> <xi:include href="dhall.section.xml" /> <xi:include href="dotnet.section.xml" /> <xi:include href="emscripten.section.xml" /> |