dream2nix/overrides/python/torch/default.nix

21 lines
540 B
Nix

{
config,
lib,
...
}: {
# stripping doesn't reduce the file size much, and it takes a long time
mkDerivation.dontStrip = true;
# use the autoAddOpenGLRunpathHook to add /run/opengl-driver/lib to the RPATH
# of all ELF files
deps = {nixpkgs, ...}: {
inherit (nixpkgs.cudaPackages) autoAddOpenGLRunpathHook;
};
mkDerivation.nativeBuildInputs = [
config.deps.autoAddOpenGLRunpathHook
];
# this file is patched manually, so ignore it in autoPatchelf
env.autoPatchelfIgnoreMissingDeps = ["libcuda.so.1"];
}