dream2nix/overrides/python/torch/default.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"];
}