nix-community/dream2nix
1
1
Fork 0
mirror of https://github.com/nix-community/dream2nix.git synced 2025-01-09 00:08:02 +03:00
Code Issues Projects Releases Wiki Activity

fix(overlays/torch): use autoAddOpenGLRunpathHook

Browse Source
This commit is contained in:
DavHau 2023-09-13 13:50:19 +02:00
parent 77eeb06927
commit 19bfd98a97
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
overrides/python/torch/default.nix
View File

@ -6,12 +6,15 @@
# stripping doesn't reduce the file size much, and it takes a long time # stripping doesn't reduce the file size much, and it takes a long time
mkDerivation.dontStrip = true; 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 # this file is patched manually, so ignore it in autoPatchelf
env.autoPatchelfIgnoreMissingDeps = ["libcuda.so.1"]; env.autoPatchelfIgnoreMissingDeps = ["libcuda.so.1"];
# patch the rpath so libcuda.so.1 can be found at /run/opengl-driver/lib
env.cudaPatchPhase = ''
patchelf $out/${config.deps.python.sitePackages}/torch/lib/libcaffe2_nvrtc.so \
--add-rpath /run/opengl-driver/lib
'';
mkDerivation.postPhases = ["cudaPatchPhase"];
} }