1d63f89caa
There are many different versions of the `cudatoolkit` and related
cuda packages, and it can be tricky to ensure they remain compatible.
- `cudaPackages` is now a package set with `cudatoolkit`, `cudnn`, `cutensor`, `nccl`, as well as `cudatoolkit` split into smaller packages ("redist");
- expressions should now use `cudaPackages` as parameter instead of the individual cuda packages;
- `makeScope` is now used, so it is possible to use `.overrideScope'` to set e.g. a different `cudnn` version;
- `release-cuda.nix` is introduced to easily evaluate cuda packages using hydra.
51 lines
1.1 KiB
Nix
51 lines
1.1 KiB
Nix
{ lib
|
|
, stdenv
|
|
, fetchurl
|
|
, autoPatchelfHook
|
|
, autoAddOpenGLRunpathHook
|
|
}:
|
|
|
|
pname:
|
|
attrs:
|
|
|
|
let
|
|
arch = "linux-x86_64";
|
|
in stdenv.mkDerivation {
|
|
inherit pname;
|
|
inherit (attrs) version;
|
|
|
|
src = assert (lib.hasAttr arch attrs); fetchurl {
|
|
url = "https://developer.download.nvidia.com/compute/cuda/redist/${attrs.${arch}.relative_path}";
|
|
inherit (attrs.${arch}) sha256;
|
|
};
|
|
|
|
nativeBuildInputs = [
|
|
autoPatchelfHook
|
|
# This hook will make sure libcuda can be found
|
|
# in typically /lib/opengl-driver by adding that
|
|
# directory to the rpath of all ELF binaries.
|
|
# Check e.g. with `patchelf --print-rpath path/to/my/binary
|
|
autoAddOpenGLRunpathHook
|
|
];
|
|
|
|
buildInputs = [
|
|
stdenv.cc.cc.lib
|
|
];
|
|
|
|
dontBuild = true;
|
|
|
|
# TODO: choose whether to install static/dynamic libs
|
|
installPhase = ''
|
|
runHook preInstall
|
|
rm LICENSE
|
|
mkdir -p $out
|
|
mv * $out
|
|
runHook postInstall
|
|
'';
|
|
|
|
meta = {
|
|
description = attrs.name;
|
|
license = lib.licenses.unfree;
|
|
platforms = lib.optionals (lib.hasAttr arch attrs) [ "x86_64-linux" ];
|
|
};
|
|
}
|