mirror of
https://github.com/nix-community/dream2nix.git
synced 2024-09-11 22:37:18 +03:00
feat(overlays/torch): add /run/opengl-driver/lib to RPATH
This allows resolving libcuda.so.1
This commit is contained in:
parent
8d5178a665
commit
77eeb06927
@ -3,6 +3,15 @@
|
||||
lib,
|
||||
...
|
||||
}: {
|
||||
env.autoPatchelfIgnoreMissingDeps = ["libcuda.so.1"];
|
||||
# stripping doesn't reduce the file size much, and it takes a long time
|
||||
mkDerivation.dontStrip = true;
|
||||
|
||||
# this file is patched manually, so ignore it in autoPatchelf
|
||||
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"];
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user